anaconda虚拟环境中报错：OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA

目录

问题说明

解决方案

第一步，打开cmd终端输入 “nvidia-smi”

第二步， 下载安装 Cuda

第三步，CUDA 环境变量配置

第四步：安装 Cudnn

参考资料

---

问题说明

在安装mmcv时，anaconda虚拟环境中显示cuda没有安装

 OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.

首先看一下cmd中的cuda版本：

nvidia-smi

根据网上资料安装cudatoolkit

conda install -c conda-forge cudatoolkit-dev

但是出现报错ValueError: unsupported format character 'T' (0x54) at index 970

又尝试在虚拟环境中采用以下代码，仍然无法解决

conda install cudatoolkit
conda install cudnn

解决方案

本机base环境中没有安装了cuda，也没有配置环境变量，需要从头安装cuda， cuda toolkit和cudnn。

第一步，打开cmd终端输入 “nvidia-smi”

如果有如下图所示的结果，就说明NVIDIA驱动安装成功了：

第二步， 下载安装 Cuda

前往网站 CUDA 下载，和上一步版本匹配的cuda，比如我是

然后按照安装提示进行安装即可

第三步，CUDA 环境变量配置

在Windows 搜索框中输入：环境变量，如下图所示：

• 自动生成的环境变量如下
• 
• 需要手动添加以下环境变量:

• CUDA_BIN_PATH=%CUDA_PATH%\bin
  CUDA_LIB_PATH=%CUDA_PATH%\lib\x64
  CUDA_SDK_BIN_PATH=%CUDA_SDK_PATH%\bin\win64
  CUDA_SDK_LIB_PATH=%CUDA_SDK_PATH%\common\lib\x64
  

• 添加上上述环境变量后,在系统环境变量Path中添加下列路径:

• %CUDA_BIN_PATH%
  %CUDA_LIB_PATH%
  %CUDA_SDK_BIN_PATH%
  %CUDA_SDK_LIB_path%
  

  在命令行中输入:nvcc --version查看是否安装成功

第四步：安装 Cudnn

首先，前往 cudnn 下载 来下载 cudnn库

然后，对cudnn 进行解压，最后将解压后的 bin，include，lib文件夹下的内容拷贝到 cuda 对应的 bin，include，lib 下即可。

ps：这一步我没有做，但是mmcv可以安装了

参考资料

一文理清CUDA环境配置（Linux/Windows，包括pytorch、mmdetection等） - 知乎

https://stackoverflow.com/questions/52731782/get-cuda-home-environment-path-pytorch

Windows 环境搭建： Nvidia驱动+Cuda+Cudnn - 知乎 (zhihu.com)