ubuntu18.04安装pytorch

---

默认CUDA已经安装，并查看其版本号

sudo nvidia-smi

一、Anaconda的下载安装

Anaconda官网主页:https://anaconda.org/
https://www.continuum.io/downloads

到官网进行下载，之后使用下面命令安装

bash Anaconda3-xxxx.sh

一直按回车，知道出现yes位置，之后会默认安装。

其中xxx就是对应的版本号

---

二、查看自己的Python版本

python

得到自己python版本号

三、安装pytorch

1.查看对应版本的pythorch并安装

官网：https://pytorch.org/get-started/locally/

我的版本对应的版本是11.1

安装代码如下：

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

注意：
1）这步容易因网速原因失败
比如：

CondaHTTPError: HTTP 000 CONNECTION FAILED for url <https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/linux-64/qt-5.12.5-hd8c4c69_1.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.


Collecting package metadata (current_repodata.json): failed

通过更换源，可以解决这个问题：

#1、备份本地配置 
cp ~/.condarc ~/.condarc.bak

sudo gedit  ~/.condarc
#2、删除原来的内容，填入以下
channels:
  - defaults
show_channel_urls: true
channel_alias: https://mirrors.tuna.tsinghua.edu.cn/anaconda
default_channels:
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/r
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
  - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2
custom_channels:
  conda-forge: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  msys2: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  bioconda: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  menpo: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  pytorch: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud
  simpleitk: https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud



#3、清除缓存
conda clean -i

2)有的教程让独立一个新的conda环境，进行pytorch安装
但是就我的实践看，容易出现问题。其原因是独立一个新的conda环境时，是将python直接复制过去的，所以一些接口路径就变了，而导致安装失败。当然你可以一个个文件查找并将路径改正。

2.测试

代码如下（示例）：

>>> import torch
>>> torch._version_
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
AttributeError: module 'torch' has no attribute '_version_'
>>> torch.__version__
'1.11.0'
>>> 

---