Ubuntu用Anaconda 安装PyTorch、torchv
2018-06-02 本文已影响0人
糊糊小鱼干
适用于服务器上多用户安装于自己目录的情况
Anaconda
安装Anaconda见此篇
CUDA
安装CUDA见此篇
查看CUDA版本
nvcc -V
Pytorch & torchvision
修改软件源为清华的源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
去Pytorch官网根据需要的版本下载安装,需要关注的有:
- OS
- Package Manager
- Python
- CUDA
本次选择如图
1按照command提示下载
conda install pytorch torchvision cuda90 -c pytorch
完成之后可在python中查看
import torch
import torchvision
print(torch.__version__)#输出PyTorch版本