Windows下用Anaconda安裝Pytorch/torchvision

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

          

 

 

 

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