linux系统搭建人工智能pytorch框架

1 下载并安装Anaconda

wget https://repo.anaconda.com/archive/Anaconda3-2019.03-Linux-x86_64.sh

sh Anaconda3-2019.03-Linux-x86_64.sh

2 建立Anaconda环境（以python3.6为例）

conda create --name py36 python=3.6

进入建立环境，检查是否安装成功

conda activate py36

python

3安装python常用工具（进入py36后安装）

pip install numpy pandas matplotlib scikit-learn scikit-image

4安装深度学习框架（进入py36后安装；以cuda9.0为例）

conda install pytorch torchvision cudatoolkit=9.0 -c pytorch  #for gpu
conda install pytorch torchvision cpuonly -c pytorch  # for cpu

5安装pytorch的计算机视觉包

pip install torchvision

 6进入py36环境测试

python

import torch
import torchvision
torch.cuda.is_available()

 输出True说明GPU可用