參考文章:https://zhuanlan.zhihu.com/p/59031211
方案介紹
操作系統:Win10
Anaconda:新版
CUDA:新版
cudnn:新版
深度學習框架:pytorch
第一步:Anaconda安裝
參考文章:
https://www.docin.com/p-2211294227.html
下載Anaconda
目前最新:Anaconda3-2019.10-Windows-x86_64
原地址:https://repo.continuum.io/archive/
清華鏡像:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
百度雲鏈接:https://pan.baidu.com/s/1SDBFoKxujlGX9IH7p0JUhA
提取碼:earb
注意事項
- windows里不要裝其他的python
- 路徑可以默認
- 在安裝的過程中勾選 PATH
檢查
- 開命令行(運行;輸入‘cmd’;回車)
- 終端輸入‘python’,出現python和anaconda表示安裝成功
第二步:創建pytorch虛擬環境
- 進入 anaconda navigator ,創建名為 pytorch 的虛擬環境。登錄pytorch官網:https://pytorch.org/get-started/locally/
- 根據自己的環境選擇對應的pytorch包
- 選擇用pip方式安裝
- 復制相應的命令,打開 Anaconda Prompt ,激活相應虛擬環境 activate mypytorch。
- 在終端中輸入:(第一句就是復制來的命令)
pip install https://download.pytorch.org/whl/cpu/torch-1.0.0-cp36-cp36m-win_amd64.whl
pip install torchvision
- 因為是在anaconda的虛擬環境中裝的,所以隨時可以廢棄。
第三步:安裝cuda
登錄官網查看適合自己顯卡的版本https://developer.nvidia.com/cuda-gpus
本人下載版本:cuda_10.1.105_418.96_win10(rtx1060)
百度雲鏈接:https://pan.baidu.com/s/1xqGB5x8Bpz1vhXJDeV4ADQ
提取碼:yk68
第四步:安裝cudnn
登錄官網 https://developer.nvidia.com/cudnn
注冊下載,直接拷到cuda路徑對應的文件夾下面就行
本人下載版本:cudnn-10.1-windows10-x64-v7.6.5.32
百度雲鏈接:https://pan.baidu.com/s/1iuWb2VSJ2I4EDCS0rDaEpQ
提取碼:6yw6