windows下安裝pytorch


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__)

繼續加油,拖更太久了!


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM