每次配置環境都令人頭疼，網上教程雖然多，但大多千篇一律，很多不詳細，或者方法不適合自己。
1 確定cudaToolkit版本
開始菜單搜索“計算機管理”→【設備管理器】→【顯示適配器】，查看是否有“NVIDIA”顯卡，如果沒有，不好意思，請安裝CPU版的Pytorch。
【桌面】→【右鍵】→【Nvidia控制面板】→【系統信息】→【組件】
2 安裝Cuda運行庫
安裝
建議你按默認安裝，這里介紹“自定義安裝”
拓展包的存儲位置，這里默認。
安裝完成后，打開【控制面板】→【程序和功能】，你會看到如下清單：
驗證。打開【cmd】，輸入nvcc -V得到如下界面，表示你安裝成功。
軟件會被強制安裝到C盤，路徑為C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\{你的版本}
3 下載cudnn深度學習加速庫
注：需要先登錄cuda才可以下載。
下載后，你會得到一個zip壓縮包，解壓，將里面文件拷貝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\{你的版本}
4 安裝PyTorch框架
我嘗試了官方源的命令行在線安裝，失敗了；更換清華大學鏡像源，也失敗了；最后選擇離線安裝，成功了！
4.1 PiP離線安裝官方源
確認版本，Python 3.8，cuda 10.1,torch 1.6, torchvision 0.7。
打開【Anconda Prompt】，輸入pip install {.whl文件路徑}安裝torch，然后安裝torchvision
驗證。打開切換到python模式，結果下圖，那么恭喜成功了。
4.2 conda離線安裝清華源
如果用迅雷/IDM下載官方源依然很慢，可以下載清華大學鏡像的，網址，得到.bz2的文件，我們用conda離線安裝，輸入命令conda install --offline {.bz2文件名}
卸載pytorch命令pip uninstall torch或者conda uninstall torch
如果本博客圖片失效，請點擊印象筆記公開鏈接