ubuntu将虚拟环境中的cuda版本加入环境变量

参考链接1: conda安装的cudatoolkit, cudnn与在主机上安装的cuda, cudnn有何关系?
参考链接2:还在系统中安装CUDA?Anaconda虚拟环境中安装CUDA配合Tensorflow-gpu-多版本

conda环境安装的cuda和cudnn跟,系统中安装的cuda和cudnn有什么区别?

conda安装仅仅服务于conda可以安装的一些pytorch/tensorflow神经网络框架所需。东西肯定是系统中安装的cuda的子集。

系统cuda安装的路径
系统安装下载的cuda一般是防止在/usr/local/cuda*下面
将cuda路径写入环境变量~/.bashrc

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH

虚拟环境中的cuda安装路径
ubuntu将虚拟环境中的cuda版本加入环境变量

使用conda管理Python环境,可以会有多个环境,每个环境的cuda版本可以不同,你可以根据你需要的cuda版本,将虚拟环境中的cuda加入到环境变量中。

加入环境变量
这里的安装路径没有单独形成cuda文件夹,都是统一存放在envs/你的虚拟环境/lib or include 文件夹下:

lib下面
在这里插入图片描述
include下面

设置CUDA_HOME 的路径为虚拟环境{enviroment}的cuda路径 anaconda/envs/{enviroment}/lib/
{enviroment}代表的是虚拟环境的名字。

然后,通过 conda 安装 cudatoolkit 包含的库文件均在 miniconda3/lib 中可以找到(或者直接在
miniconda3 / pkgs / cudatoolkit-10.x.xxx /lib 中查看),

猜你喜欢

转载自blog.csdn.net/weixin_37707670/article/details/121451114