1、安裝anaconda3:
https://www.anaconda.com/download/#windows
選Python3.6的x64版本下載、安裝。
安裝之后,把“C:\ProgramData\Anaconda3;C:\ProgramData\Anaconda3\Library\bin”加到系統環境變量(可選,主要為了后面運行conda命令)。
2、安裝 CUDA。注意:需要有Nv顯卡,否則沒必要裝了。目前最新的9.1版本:
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10
下載network或local包都可以安裝 。
3、下載 安裝anaconda下的Pytorch包:
參考 蒲嘉宸 大神的文章:https://zhuanlan.zhihu.com/p/26871672
下載地址:https://pan.baidu.com/s/1dF6ayLr#list/path=%2F
根據CUDA是7、8、9選擇 帶有cu70、cu80、cu90的最新安裝包下載。我選的是pytorch-0.3.1-py36_cuda90_cudnn7he774522_2.tar.bz2
下載完之后,運行:
cd "下載包的路徑" conda install numpy mkl cffi conda install --offline pytorch????.tar.bz2
完成。可以運行代碼進行測試。