PyTorch安装CUDA版本的正确方式

最新推荐文章于 2025-05-16 18:56:00 发布

原创最新推荐文章于 2025-05-16 18:56:00 发布 · 645 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习 #python

安装PyTorch的CUDA版本时，需要确保从官方仓库检查是否有对应版本的软件包。例如，对于CUDA10.2，应安装torch1.10.2+cu102、torchvision0.11.3+cu102和torchaudio0.10.2+cu102。如果遇到依赖不兼容问题，可能需要调整版本号以达到匹配。

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

PyTorch安装CUDA版本的正确方式

很多人按照PyTorch官方的说明安装CUDA版本，装完却发现装的是CPU版本。

原因就是是仓库里就没有这个版本的软件包，安装前需要在下面的网址查看一下儿是否有编译好的软件包：

https://download.pytorch.org/whl/torch_stable.html
https://download.pytorch.org/whl/cu102/torch_stable.html

以安装PyTorch CUDA10.2版本为例，先安装torch，发现仓库里只有torch的1.10.2版本。
再安装torchvision和torchaudio，如后两个有依赖不兼容，可以调整小版本号。
示例：

pip install torch==1.10.2+cu102 torchvision==0.11.3+cu102 torchaudio==0.10.2+cu102 -f https://download.pytorch.org/whl/torch_stable.html