Ubuntu16.04安装pytorch和torchvision，版本对应安装

最新推荐文章于 2024-07-22 11:08:20 发布

机械小蛟龙

最新推荐文章于 2024-07-22 11:08:20 发布

阅读量579

点赞数 1

分类专栏：笔记

本文链接：https://blog.csdn.net/weixin_48646621/article/details/114576551

版权

笔记专栏收录该内容

12 篇文章 1 订阅

订阅专栏

去官网下载：链接
根据CUDA的版本下载对应的pytorch和torchvision

# CUDA 9.2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=9.2 -c pytorch

# CUDA 10.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

# CUDA 10.2
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.2 -c pytorch

# CUDA 11.0
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch

# CPU Only
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cpuonly -c pytorch

直接终端下载很慢，建议复制链接去window下载，再回到ubuntu安装