求助解决CUDA运行时未找到错误:PyTorch与CUDA版本适配问题
解决CUDA runtime未找到的问题
问题根源
你安装的PyTorch 2.0.0是通过cu117索引编译的版本,仅兼容CUDA 11.7,但当前系统安装的是CUDA 12.2,版本不匹配导致报错。
具体解决步骤
选项1:升级PyTorch适配CUDA 12.x
- 卸载现有不兼容的PyTorch组件:
pip3 uninstall -y torch torchvision torchaudio
- 安装支持CUDA 12.x的PyTorch版本(兼容12.2):
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
选项2:降级CUDA适配现有PyTorch
如果想保留PyTorch 2.0.0,需:
- 卸载当前CUDA 12.2及对应CuDNN
- 重新安装CUDA Toolkit 11.7和匹配版本的CuDNN
- 将CuDNN的
bin、include、lib文件夹内的文件,分别复制到CUDA安装目录的同名文件夹中
验证配置
打开Python终端执行以下代码,输出True则表示配置成功:
import torch print(torch.cuda.is_available())
环境变量检查
确认系统环境变量设置:
CUDA_HOME指向正确的CUDA安装路径(如C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.2或v11.7)PATH中添加%CUDA_HOME%\bin和%CUDA_HOME%\libnvvp路径
内容的提问来源于stack exchange,提问作者Aaryan Kandiah
相关产品推荐
相关产品推荐

