(1)版本查看
查看cuda版本cat /usr/local/cuda/version.txt
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
可以进去修改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__)
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:linux中anaconda环境下pytorch的安装(conda安装本地包) - Python技术站