Ubuntu 下Anaconda 安装 Pytorch

在线安装

Pytorch安装

1、Anaconda 下安装Pytorch，官方的命令是

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0 -c pytorch

但是报错如下

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://conda.anaconda.org/pytorch/linux-64/pytorch-1.2.0-py3.6_cuda10.0.130_cudnn7.6.2_0.tar.bz2>
Elapsed: -

An HTTP error occurred when trying to retrieve this URL.
HTTP errors are often intermittent, and a simple retry will get you on your way.

ChunkedEncodingError(ProtocolError('Connection broken: OSError("(104, \'ECONNRESET\')")', OSError("(104, 'ECONNRESET')")))

原因主要是pytorch 和torchvision 下载失败。

2、因此可以切换清华的源：方法参考

https://blog.csdn.net/ytusdc/article/details/89469161

3、切换之后上述命令改为

conda install pytorch==1.2.0 torchvision==0.4.0 cudatoolkit=10.0

注意：必须去掉 -c pytorch ，这个参数指定的是使用官方的源进行下载

torchvision安装：

用命令 pip install torchvision 安装时候。总是自动安装pytorch

因此使用下面的命令，会只安装torchvision
 

pip install --no-deps torchvision

离线安装：

pip install torch-1.0.1-cp36-cp36m-linux_x86_64.whl 

Pytorch 版本查看安装路径查看

python 环境下，torch 和 torchvision 命令相同

>>> import torch
>>> torch.__version__
'1.1.0'
>>> torch.__path__
['/home/cv/SoftWare/Anaconda3/envs/py36/lib/python3.6/site-packages/torch']
>>>