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官网根据需要的版本下载安装,需要关注的有:

本次选择如图

1

按照command提示下载

conda install pytorch torchvision cuda90 -c pytorch

完成之后可在python中查看

import torch
import torchvision
print(torch.__version__)#输出PyTorch版本
上一篇下一篇

猜你喜欢

热点阅读