不同项目所使用的cuda版本也可能不同,网上一般方法是将环境变量提前(https://blog.csdn.net/zhaotun123/article/details/99680151),来实现cuda版本更换。本文介绍了一种快速切换cuda版本的办法:即写一个脚本从而实现进入虚拟环境自动切换cuda版本。
之前有人出的教程,但是太难找了,所以留个档自用。
下载cuda
(随便找个教程)CUDA的下载与安装_cuda下载_疯狂的小强呀的博客-CSDN博客
编写脚本
找到要切换cuda版本的项目环境(D:\Software\Anaconda3\envs\HRNetFace)
在该目录下创建如下文件:
etc内配置如下(conda、activate.d、deactivate.d均为文件夹名,env_vars.bat为.bat文件):
...
-- DLLs
-- etc
|-- conda
| |-- activate.d
| | |-- env_vars.bat
| |-- deactivate.d
| | |-- env_vars.bat
-- include
...
activate.d文件夹中的env_vars.bat:
@set CUDA_PATH=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin @set CUDA_NVVP=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp @set CUDA_lib=C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64 @set OLD_PATH=%PATH% @set PATH=%CUDA_PATH%;%CUDA_NVVP%;%CUDA_lib%;%PATH%;
将CUDA_PATH、CUDA_NVVP、CUDA_lib设置为所需cuda的bin、libnvvp、lib\x64对应绝对路径。例如我这个项目需要cuda10.0,这里即为cuda10.0的相关文件的绝对路径。
deactivate.d文件夹中的env_vars.bat:
@set PATH=%OLD_PATH%
然后就ok了