6. Ubuntu安装pytorch

lengmo1996

已于 2022-03-03 10:04:01 修改

阅读量632

点赞数

分类专栏：深度学习环境配置文章标签： Ubuntu 环境配置 pytorch

于 2022-03-03 10:03:48 首次发布

本文链接：https://blog.csdn.net/qq_33401821/article/details/123246793

版权

本文详细介绍了在Ubuntu操作系统中，如何通过conda和pip两种方式安装PyTorch的v1.8.0及v1.7.1版本，覆盖了不同版本和包管理工具的安装步骤。

摘要由CSDN通过智能技术生成

6.1 v1.8.0+conda

# CUDA 10.2
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch

# CUDA 11.1
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge

# CPU Only
conda install pytorch==1.8.0 torchvi