1.選擇對(duì)應(yīng)的版本
在運(yùn)行之前先查看GPU的使用情況（CUDA的版本）：
$ nvidia-smi # 查看GPU此時(shí)的使用情況
1.1 根據(jù)cudatoolkit的版本憾儒，在如下網(wǎng)站上選擇對(duì)應(yīng)的Pytorch版本
https://pytorch.org/get-started/previous-versions/
1.2
在官網(wǎng)https://pytorch.org/上選擇對(duì)應(yīng)的版本：
直接復(fù)制最后的命令行：
（1）conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
（2）conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install torchvision
1.3 成功安裝的方法
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
conda install pytorch torchvision cudatoolkit=10.1
conda install cudatoolkit
conda uninstall pytorch
2. 離線安裝
現(xiàn)在命令窗口執(zhí)行在線安裝的命令屁桑，然后會(huì)如下圖給出“Downloading ----”亡驰，復(fù)制鏈接攀涵，在瀏覽器打開就可以離線下載對(duì)應(yīng)版本了。
還需要安裝對(duì)應(yīng)版本的cudnn和cudatoolkit；
conda install cudnn=7.3.1
conda install cudatoolkit=10.1
https://repo.anaconda.com/pkgs/main/linux-64/cudnn-7.6.4-cuda10.0_0.tar.bz2
https://repo.anaconda.com/pkgs/main/linux-64/cudatoolkit-10.0.130-0.tar.bz2
E:\谷歌瀏覽器
2.1 Linux版本
安裝之前，需要安裝numpy拗秘，conda install numpy；版本為1.17
(1) 下載地址：
https://download.pytorch.org/whl/cu101/torch-1.3.0-cp36-cp36m-manylinux1_x86_64.whl
（2）安裝命令
pip install /home/pengbang/software/torch-1.3.0-cp36-cp36m-manylinux1_x86_64.whl
2.2 Windows版本
(1) 下載地址：
https://download.pytorch.org/whl/cu101/torch-1.3.0-cp36-cp36m-win_amd64.whl
(2) 安裝命令
pip install E:\谷歌瀏覽器\torch-1.3.0-cp36-cp36m-win_amd64.whl
pip install E:\谷歌瀏覽器\torch-1.0.1-cp36-cp36m-win_amd64.whl （成功祈惶，cudatoolkit=9.0的對(duì)應(yīng)版本）
2.3 更新numpy
pip install --upgrade numpy
2. 查詢是否安裝了CPU版本
當(dāng)前環(huán)境下的安裝包版本：
cudatoolkit 9.0 1 defaults
cudnn 7.1.4 cuda9.0_0 defaults
numpy 1.17.3 pypi_0 pypi
torch 1.0.1 pypi_0 pypi
torchvision 0.4.1 pypi_0 pypi
3. 檢查安裝是否成功
import torch
print(torch.cuda.is_available())
輸出True即可雕旨。
或者
import torch
print(torch.version.cuda)
輸出CUDA的版本號(hào)即可。
此時(shí)運(yùn)行你的深度學(xué)習(xí)程序應(yīng)該就可以成功啦行瑞。
參考資料
[1] https://pytorch.org/
[2] windows10環(huán)境下安裝深度學(xué)習(xí)環(huán)境anaconda+pytorch+CUDA+cuDDN
[3] PyTorch中文教程
[4] Win10系統(tǒng)安裝GPU版PyTorch
