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

38 篇文章 1 订阅
19 篇文章 2 订阅

参考链接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 中查看),

评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值