cuda torch torchvision 版本对应

最新推荐文章于 2025-03-10 12:30:31 发布

临街的小孩

最新推荐文章于 2025-03-10 12:30:31 发布

阅读量3.7k

点赞数 1

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

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

版权

人工智能专栏收录该内容

20 篇文章

订阅专栏

# 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