linux中anaconda环境下pytorch的安装(conda安装本地包)

跑代码的时候遇到和这位博主几乎一模一样的问题,安装的也是同一版本。目前清华源已经停止服务,如果要自己下载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__)

    原文作者:pytorch
    原文地址: https://www.cnblogs.com/youyou0/p/10909491.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞