跑代碼的時候遇到和這位博主幾乎一模一樣的問題,安裝的也是同一版本。目前清華源已經停止服務,如果要自己下載pytorch包的話估計只能在官網下載了。
原文:https://blog.csdn.net/summer2day/article/details/88652934
pytorch的安裝
(1)版本查看
查看cuda版本cat /usr/local/cuda/version.txt
(1)版本查看
查看cuda版本cat /usr/local/cuda/version.txt
查看cudnn版本cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A
去官網找到適合的版本用conda安裝
http://webcache.googleusercontent.com/search?q=cache:https://pytorch.org/get-started/previous-versions/
(python版本,64或者32位,cuda,cudnn的版本)
(2)設置鏡像
因為Anaconda.org的服務器在國外。可以用清華TUNA鏡像源,包含Anaconda倉庫的鏡像,將其加入conda的配置。否則速度會很慢並且經常失敗,其中出現”CondaHTTPError: HTTP None None for url……”這類錯誤時。
# 添加Anaconda的TUNA鏡像
$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 設置搜索時顯示通道地址
$ conda config --set show_channel_urls yes
http://webcache.googleusercontent.com/search?q=cache:https://pytorch.org/get-started/previous-versions/
(python版本,64或者32位,cuda,cudnn的版本)
(2)設置鏡像
因為Anaconda.org的服務器在國外。可以用清華TUNA鏡像源,包含Anaconda倉庫的鏡像,將其加入conda的配置。否則速度會很慢並且經常失敗,其中出現”CondaHTTPError: HTTP None None for url……”這類錯誤時。
# 添加Anaconda的TUNA鏡像
$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
$ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# 設置搜索時顯示通道地址
$ conda config --set show_channel_urls yes
123456
可以進去修改vim ~/.condarc
(4)開始安裝
例如我的是conda install pytorch=0.3.1 cuda90 -c pytorch
conda list 可以查詢你現在安裝了哪些庫
但是!!!我設置了鏡像源還是從官網上下的不知道怎么回事,於是我只好去清華鏡像源網站上自己找好要下載的包,然后離線安裝
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
conda install --offline XXXX…tar.bz2
(5)查看是否安裝成功
安裝成功后進入python環境可以查看pytorch版本:
import torch
print(torch.__version__)
可以進去修改vim ~/.condarc
(4)開始安裝
例如我的是conda install pytorch=0.3.1 cuda90 -c pytorch
conda list 可以查詢你現在安裝了哪些庫
但是!!!我設置了鏡像源還是從官網上下的不知道怎么回事,於是我只好去清華鏡像源網站上自己找好要下載的包,然后離線安裝
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
conda install --offline XXXX…tar.bz2
(5)查看是否安裝成功
安裝成功后進入python環境可以查看pytorch版本:
import torch
print(torch.__version__)