win10安裝CUDA9.0+CUDANN+Anaconda3+PYTORCH1.x

當前神經網絡方法的訓練和預測越來越依賴於GPU並行運算,因此在win10上配置CUDA+Pytorch,記錄配置過程。

硬件平臺:

    win10(64位), NVIDIA Geforce 940M

軟件下載:

    1:CUDA9.0,安裝時注意加載路徑 (環境變量)

官網地址:CUDA9.0下載

    2:CUDANN,對應CUDA9.0版本的CUDANN

官網地址:CUDANN7.6下載

    3:解壓CUDANN

    將對應bin/cudann64_7.dll, include/cudnn.h, lib/x64/cudnn.lib分別加載到CUDA包對應的bin, include, lib/x64文件中;

打開控制檯,輸入:

nvcc -V

顯示如下:

    4:安裝Anaconda3.5.2.0(對應Python3.6),清華鏡像下載

    官網地址:Anaconda3.5.2.0下載

    安裝環境時添加python路徑,正確安裝後,打開控制檯顯示結果(注意有anaconda後綴):

     5:安裝pytorch

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch

    6:設置jupyter notebook默認地址

    找到jupyter notebook屬性頁:

    將目標值設爲:

C:\ProgramData\Anaconda3\python.exe C:\ProgramData\Anaconda3\cwp.py C:\ProgramData\Anaconda3 C:\ProgramData\Anaconda3\python.exe C:\ProgramData\Anaconda3\Scripts\jupyter-notebook-script.py "設置的文件路徑"

     起始位置與“設置的文件路徑”相同

    代碼補全:

pip install jupyter_contrib_nbextensions
jupyter contrib nbextension install --user
pip install--user jupyter_nbextensions_configurator
jupyter nbextensions_configuratorenable--user

    出現結果(勾選Hinterland):

     7:測試

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章