跑代码的时候遇到和这位博主几乎一模一样的问题,安装的也是同一版本。目前清华源已经停止服务,如果要自己下载pytorch包的话估计只能在官网下载了。
原文:https://blog.csdn.net/summer2day/article/details/88652934
pytorch的安装
(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
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__)