【杂记】给不同python虚拟环境配置不同CUDA环境

配置CUDA

目标：当前系统默认的CUDA是10.2的，想给指定虚拟环境配置10.0的CUDA

conda

在环境目录创建两个文件

activate.d\env_vars.bat：（这里配置的是10.0，其他同理）

@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%;

deactivate.d\env_vars.bat：

@set PATH=%OLD_PATH%

这样在cmd里activate环境，对应python里的cuda就是这个路径的了。

在这里插入图片描述

pycharm

菜单栏->运行->编辑配置->配置环境变量

在这里插入图片描述

通过代码

import os
os.environ["CUDA_PATH"] = r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin"
os.environ["CUDA_NVVP"] = r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\libnvvp"
os.environ["CUDA_lib"] = r"C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64"

可以用这个查看环境是否被修改：

for name,path in os.environ.items():
    print(name,':',path)

0 条评论