cuda torch torchvision 版本对应

最新推荐文章于 2025-06-13 19:20:29 发布

临街的小孩

最新推荐文章于 2025-06-13 19:20:29 发布

阅读量3.8k

点赞数 1

CC 4.0 BY-SA版权

分类专栏：人工智能文章标签： pytorch 深度学习 python

本文链接：https://blog.youkuaiyun.com/W25679/article/details/127146671

人工智能专栏收录该内容

22 篇文章

订阅专栏

本文介绍如何使用conda安装指定CUDA版本的PyTorch及其相关库，包括torchvision和torchaudio。覆盖CUDA10.2、CUDA11.3、CUDA11.6及仅CPU版本。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

# CUDA 10.2
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=10.2 -c pytorch
# CUDA 11.3
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.3 -c pytorch
# CUDA 11.6
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
# CPU Only
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cpuonly -c pytorch

网址 Previous PyTorch Versions | PyTorch