进入cmd,输入
cd C:\ProgramData\Anaconda3\envs\ env # Anaconda 默认安装在C盘此目录下 env:为所需切换cuda版本的虚拟环境
cd /d D:\Anaconda3\envs\ env #Anaconda 安装在其他位置
mkdir .\etc\conda\activate.d #创建activate.d文件夹
mkdir .\etc\conda\deactivate.d #创建deactivate.d文件夹
#activate.d
echo @echo off > .\etc\conda\activate.d\activate_cuda10.bat
echo set OLD_PATH=%%PATH%% >> .\etc\conda\activate.d\activate_cuda10.bat
echo set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
echo set CUDA_NVVP=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp
echo set CUDA_lib=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64
echo set OLD_PATH=%PATH%
echo set PATH=%CUDA_PATH%;%CUDA_NVVP%;%CUDA_lib%;%PATH%;
#deavtivate.d
echo @echo off > .\etc\conda\deactivate.d\deactivate_cuda10.bat
echo set PATH=%%OLD_PATH%% >> .\etc\conda\deactivate.d\deactivate_cuda10.bat
手动设置cuda,在cmd或pycharm终端输入如下:
call D:\anaconda3\envs\ssl_ecg\etc\conda\activate.d\activate_cuda10.bat
查看cuda版本
nvcc -V