【更新時間2020.5.7】pytorch 1.5+win10 +python3.7+cuda10.2 安裝

【更新時間2020.5.7】pytorch 1.5+win10 +python3.7+cuda10.2 安裝

 

點擊進入官網 https://pytorch.org/

pip安裝

Python3.7+pip安裝cpu版本

pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html

Python3.7+pip安裝gpu版本 
目前gpu版本支持cuda8.0,cuda9.0和cuda10.0，請選擇對應的版本下載安裝，不要同時執行下面三個命令！喜歡conda的可以選擇

 

pip install torch===1.5.0 torchvision===0.6.0 -f https://download.pytorch.org/whl/torch_stable.html

conda安裝

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

 

不得不說，比tensorflow 的安裝方便了1000倍！

cuda10.2

https://developer.nvidia.com/cuda-downloads

https://developer.download.nvidia.com/compute/cuda/10.2/Prod/local_installers/cuda_10.2.89_441.22_win10.exe

cudnn7.65

https://developer.nvidia.com/rdp/cudnn-download

https://developer.nvidia.com/compute/machine-learning/cudnn/secure/7.6.5.32/Production/10.2_20191118/cudnn-10.2-windows10-x64-v7.6.5.32.zip

檢測是否成功！

import torch

print(torch.cuda.is_available())

如果返回True則可以使用，經測試可以正常使用

還有問題的，可以加羣264191384 註明（pytorch）