windows下安裝torch=1.2.0環境
安裝內容:
torch:1.2.0
torchvision:0.4.0
一、安裝anaconda
首先安裝anaconda,最新版的anaconda默認安裝python3.7,不過一般不要用最新的,因為一些包的更新可能趕不上。因此考慮裝python3.6。請參考:
https://www.cnblogs.com/zhangkanghui/p/10756630.html
二、安裝CUDA和Cudnn
CUDA:NVIDIA的顯卡驅動程序和CUDA完全是兩個不同的概念哦!CUDA是NVIDIA推出的用於自家GPU的並行計算框架,也就是說CUDA只能在NVIDIA的GPU上運行,而且只有當要解決的計算問題是可以大量並行計算的時候才能發揮CUDA的作用。CUDA的本質是一個工具包(ToolKit)用於GPU加速運算,安裝完成之后可以查看一下C:\Program Files\NVIDIA GPU Computing Toolkit路徑下,或者使用命令行進行查詢。
Cudnn:cuDNN是一個SDK,是一個專門用於神經網絡的加速包,一般是要與CUDA版本匹配才能使用。
安裝內容——torch:1.2.0,torchvision:0.4.0需要搭配的環境是CUDA和Cudnn分別為CUDA10和Cudnn7.4.1.5
cuda10官網下載地址:
https://developer.nvidia.com/cuda-10.0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
Cudnn7.3.1.5下載地址:
鏈接:https://pan.baidu.com/s/1SoDhOJj55Xu63l6U17eWaw
提取碼:vci6
下載完成之后得到這兩個文件:
注意這里選擇自定義,否者可能會安裝失敗
安裝完成之后,找到這個文件夾C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0
然后大家把Cudnn的內容進行解壓。直接復制到這個文件夾下就可以了!
三、安裝pytorch
pytorch的官方安裝方法:
# CUDA 10.0
pip install torch===1.2.0 torchvision===0.4.0 -f https://download.pytorch.org/whl/torch_stable.html
# CUDA 9.2
pip install torch==1.2.0+cu92 torchvision==0.4.0+cu92 -f https://download.pytorch.org/whl/torch_stable.html
這種官方安裝放法可能會很慢,或者中斷失敗現象,所以建議使用第二種安裝方法,離線安裝:
https://download.pytorch.org/whl/torch_stable.html
這個里面的cp表示的就是python版本,我裝的是3.6所以是圈起來的這兩個
然后使用迅雷X下載就好了,下載完成找到相應的文件夾,在路徑框輸入cmd打開命令行:
pip install torch-1.2.0-cp36-cp36m-win_amd64.whl
pip install torchvision-0.4.0-cp36-cp36m-win_amd64.whl
或者使用百度網盤下載:
鏈接:https://pan.baidu.com/s/1rwT0QGdT5Tpyq6Tg9yACkQ
提取碼:gcmt
安裝完成之后執行以下命令:
import torch
print(torch.__version__)
繼續加油,拖更太久了!