Ubuntu利用Anaconda3安装Pytorch

提前：已经完成Anaconda3的安装

1.建立虚拟环境

conda create -n your_name(自己命令) python=3.X

2.激活虚拟环境

activate your_name

3.安装第三方依赖环境，如numpy,mkl,cffi

conda install numpy mkl cffi

 

4.进入官网，找到合适自己的版本及命令

丢个链接：https://pytorch.org/

根据自己的需求选择，第一行为稳定版本和测试版本；

第二行选择系统，一般会自动检测到当前系统；

第三行选择安装包，如果是C++目前只能用LibTorch安装；

第四行选择语言；

第五行选择CUDA版本；

最后会给出安装指令，按照给出的指令安装即可。

例如上图命令所示，在命令行输入：

conda install pytorch torchvision cudatoolkit=10.2 -c pytorch

5.通过python import验证是否安装成功

打开python，输入

import torch

点击回车，无输出，则安装成功。

6.查看pytorch版本

打开python，输入命令如下：

import torch
print(torch.__version__)