pytorch-gpu安装

最新推荐文章于 2025-04-04 10:48:25 发布

weixin_39154610

最新推荐文章于 2025-04-04 10:48:25 发布

阅读量416

点赞数

文章标签： pytorch 人工智能 python

原文链接：https://blog.csdn.net/qq_62928482/article/details/139708830

版权

1.确定CUDA版本

在命令提示符中输入nvcc -V，查看CUDA版本

2.确定python版本

我从gpt上面查出，pytorch2.4.0是支持python3.8到python3.12的
大家也可以直接去pytorch官网看，对应的python版本
https://github.com/pytorch/vision#installation

3.确定pytorch版本

进入Pytorch官网：https://pytorch.org/get-started/previous-versions/

我这里用相对较新的版本2.4.0。我的cuda是11.8，所以我选择以上版本

4.安装pytorch

确定好版本后，使用相应的安装命令，拷贝到终端进行安装

# CUDA 11.8
pip install torch==2.4.0 torchvision==0.19.0 torchaudio==2.4.0 --index-url https://download.pytorch.org/whl/cu118

5.检验是否安装成功

执行以下命令，若返回相应的结果，说明GPU版PyTorch安装成功！

python
import torch
torch.cuda.is_available()

6.错误处理

错误全称：OSError: [WinError 126] 找不到指定的模块。 Error loading “D:\Program Files\Python\Python311\Lib\site-packages\torch\lib\fbgemm.dll” or one of its dependencies.

解决：在路径“D:\Program Files\Python\Python311\Lib\site-packages\torch\lib”中加入下列网址下载的libomp140.x86_64.dll。(下载资源)
https://www.dllme.com/dll/files/libomp140_x86_64/00637fe34a6043031c9ae4c6cf0a891d/download
原文链接：https://blog.csdn.net/marujie123/article/details/141094387