参考链接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}代表的是虚拟环境的名字。
vim ~/.bashrc
#例如
export PATH=/home/lus/miniconda3/envs/open-mmlab/lib/:$PATH
export PATH={enviroment_path}/lib/:$PATH
{enviroment_path}是你自己的路径
然后,通过 conda 安装 cudatoolkit 包含的库文件均在 miniconda3/lib 中可以找到(或者直接在
miniconda3 / pkgs / cudatoolkit-10.x.xxx /lib 中查看),