1. 安裝Anaconda最新版
2. 依賴的環境
Anaconda3 x64 (with Python 3.5/3.6)
Windows 64位系統(Windows 7 或 Windows Server 2008 及以上)
CUDA 8 / CUDA 9
cuDNN v5以上
如果安裝了CUDA編譯的包,請確保你的電腦有Nvidia的顯卡。
注:這裏沒有介紹GPU版本的安裝方法,如需要的請搜索其他博文。
安裝CUDA 8
(1)首先在官網上(https://developer.nvidia.com/cuda-downloads)下載CUDA,選擇自己合適的版本。
該鏈接界面只顯示最新版本。若需要下載以前的版本,可在頁面下方點擊Legacy Releases,選擇自己需要的其他版本,這裏安裝的是cuda8.0。
這裏寫圖片描述
在Windows中使用迅雷下載,這樣比較快。
(2)根據相應的系統選項,我選擇的是cuda_8.0.61_win10.exe(大小爲1.3G),安裝的時候建議選擇 自定義 而不是“精簡”(從下面的英文解釋可以看出,其實這裏的精簡寫成完整應該更貼切,他會安裝所有組件並覆蓋現有驅動,然而我並不想安裝全家桶,何況我的官方顯卡驅動比他的新)。
雙擊安裝下載的.exe文件,然後選擇解壓路徑,如下圖,解壓到哪裏無所謂,安裝成功會自動刪除;
解壓完成後,得到如下圖:
- 精簡:安裝所有CUDA模塊,並覆蓋掉當前的NVIDIA驅動程序;(說實話,容易出問題)
- 自定義:選擇自己想要安裝的模塊,此處選擇這裏;
選擇自定義後,出現下圖所示:
下面幾個模塊準確具體有什麼用,不能100%確定,但能大概纔出來:
- CUDA:這個是必須的,下面有CUDA Runntime、Samples一些東西;
- NVIDIA GeForce Experience:這個好像是爲了更好的遊戲體驗,之前安裝顯卡驅動程序時也提示是否安裝,果斷拒絕了;
- Other components:這裏的PhysX好像也是爲了遊戲體驗來的;
- Driver components:這個就要慎重了,意思就是重新安裝顯卡驅動程序;如果之前已經成功安裝驅動程序,這裏就不用選了;如果之前沒安裝驅動程序,建議還是去官網上單獨下載驅動程序進行安裝吧;
選擇好需要安裝的模塊,這裏我都選了,但是第一項cuda中的 Visual Studio 的選項,不用的話,好像就不用安裝。
選擇安裝路徑了,我的選擇如下圖:
- 在如圖所示位置建立相應文件夾,然後再指定安裝路徑;
安裝成功後;Ctrl+R,打開cmd:
nvcc -V
輸出版本信息,則表明安裝成功。
配置環境變量
將F:\CUDA\NVIDIA GPU Computing Tookit\CUDA\v8.0\lib\x64
添加的系統變量的path中;
2. 安裝CuDNN
下載對應CUDA 8.0版本的CuDNN:下載鏈接
(如果安裝的是其他版本的CUDA,注意CuDNN的版本)
下載完成後,解壓得到一個名爲cuda的文件夾;將該文件夾下的文件複製到上一步安裝的CUDA中;注意對應的文件夾;
./cuda/bin/**.dll 複製到 ./NVIDIA GPU Computing Tookit/CUDA/v8.0/bin/
./cuda/include/**.dll 複製到 ./NVIDIA GPU Computing Tookit/CUDA/v8.0/include/
./cuda/lib/x64/**.dll 複製到 ./NVIDIA GPU Computing Tookit/CUDA/v8.0/lib/x64/
安裝完成;
3. 測試1
使用VS2013打開./cuda/v8.0/Samples_vs2013.sln;
加載完成後,執行本地Windows編譯,最後輸出成功,見下圖;
3. 開始菜單打開Anaconda Prompt,在裏面輸入conda create -n pytorch python=3.5,爲pytorch創建一個虛擬環境
安裝cuda參考:
https://blog.csdn.net/weixin_34409703/article/details/93226830
https://blog.csdn.net/yimingsilence/article/details/79126914
https://blog.csdn.net/zhuiqiuzhuoyue583/article/details/88756053
安裝Pytorch/torchvision參考:
https://blog.csdn.net/yimingsilence/article/details/79126914
https://blog.csdn.net/Tan915730/article/details/78954482
https://blog.csdn.net/zhuiqiuzhuoyue583/article/details/85327559
https://blog.csdn.net/hpuspring/article/details/79756211
https://zhuanlan.zhihu.com/p/37748231