Ubuntu16.04 conda 安装 pytorch

1.基本概念

1.1 Anaconda

Anaconda是专注于数据分析的Python发行版本，包含了conda、Python等190多个科学包及其依赖项。我们本篇要介绍的anaconda虚拟环境管理就是使用其中的conda。conda是一款极为特殊的环境管理工具，之所以说它特殊，conda的设计思想——conda将几乎所有的工具、第三方包都当做package对待，甚至包括python和conda自身！因此，conda打破了包管理与环境管理的约束，能非常方便地安装各种版本python、各种package并方便地切换。也就是说，在进行虚拟环境管理时，conda可以独立于系统原有的Python版本而创建任意版本的Python解释器，这是virtualenv、virtualenvwrapper等工具所不具备的。

在Anaconda中conda可以理解为一个工具，也是一个可执行命令，其核心功能是包管理与环境管理。所以对虚拟环境进行创建、删除等操作需要使用conda命令:

(1). conda list：查看安装了哪些包。
(2). conda install package_name(包名)：安装包
(3). conda env list 或 conda info -e：查看当前存在哪些虚拟环境
(4). conda update conda：检查更新当前conda

1.2 Pytorch

安装 PyTorch 会安装两个模块, 一个是 torch, 一个torchvision, torch 是主模块, 用来搭建神经网络的, torchvision是辅模块, 有数据库, 还有一些已经训练好的神经网络（(VGG, AlexNet, ResNet)）可以直接用。.
(1) 查看Python版本号
命令行下输入python

python -V  #注意V是大写

2.Pytorch安装准备

查看cuda版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

查看Anaconda内置的Python

激活anaconda环境
conda activate
查看Python版本号
python -V  #注意V是大写

2.1 创建虚拟环境

删去安装失败的虚拟环境。conda remove -n pytorchabc –all (pytorchabc是环境名字)

Conda create –n pytorchD python=3.6.5 

pytorchD是环境名字，指定了python版本

提示：一系列包被下载和安装
包含：openssl、python、tk、whel等包。

Preparing\verifying\Executing 完成。表示环境创建结束

添加清华镜像：

得到安装命令

conda install pytorch torchvision cudatoolkit=10.1 –c pytorch

2.2 准备安装

如果使用上面得到的命令直接安装，则会出现包下载失败的问题。即使tsinghua镜像可以使用也不行。
那么，就考虑先手动下载，然后拷贝到pkgs目录后，再解压安装。
具体步骤如下：

	conda info找到 pkgs目录
/home/zhangsan/anaconda3/pkgs

手动从清华镜像下载到 /home/ zhangsan/tools/ 下面

https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

然后解压拷贝到各自目录下

cp –r  /home/ zhangsan/tools/*  /home/zhj/anaconda3/pkgs/

2.3 开始安装

输入命令

conda install pytorch torchvision cudatoolkit=10.1 –c pytorch

通过环境检测之后，会需要下载个别的包。

`

2.4 安装测试

进入python

python
import torch

conda install matloplib

use the following channel
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/

reference:
https://blog.csdn.net/weixin_44210775/article/details/88758575
https://www.cnblogs.com/chenhuabin/p/10718471.html https://zhuanlan.zhihu.com/p/73741240