RuntimeError: cuDNN error: CUDNN_STATUS_EXECUTION_FAILED
原因:pytorch与cuda版本不对
(也有可能是内存空间不足,可以更改虚拟空间大小)
先卸载了pytorch:conda uninstall pytorch,再安装cuda的话,会自动覆盖cuda版本。
打开cmd,命令行输入
import torch
print(torch.__version__)
print(torch.version.cuda)
若安装的cuda版本与torch对应版本不对,则出现类似错误。
以下是安装cuda:
1.打开NVIDIA控制面板,查看当前显卡驱动能够支持的CUDA版本:
2.下载cuda地址
https://developer.nvidia.com/cuda-toolkit-archive
或者安装离线包https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/中下载所需的.tar.bz2
后台conda install XXXX..tar.bz2
安装完成后还要安装
首先anaconda conda 切换为国内源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
# 设置搜索时显示通道地址
conda config --set show_channel_urls yes
命令行输入: conda install pytorch torchvision cudatoolkit=10.0
安装其他的包
pytorch官网:Pytorch官网
根据实际情况下载:
3.下载成功后,双击exe文件安装。
安装成功的验证方法是在cmd下输入nvcc -V
安装成功.在系统变量可看到:
或者在安装路径下可看到nvcc.exe