Ubuntu用Anaconda 安装PyTorch、torchvision

适用于服务器上多用户安装于自己目录的情况

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版本
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容