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 中查看),

  • 7
    点赞
  • 34
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
要在Ubuntu虚拟环境安装CUDA,您可以按照以下步骤进行操作: 1. 首先,确保您的系统满足CUDA的硬件和软件要求。您可以在NVIDIA官方网站上查找适用于您的显卡的CUDA版本和相应的驱动程序。 2. 在安装CUDA之前,您需要安装一些依赖项。打开终端并运行以下命令更新软件包列表: ``` sudo apt update ``` 3. 安装必要的依赖项,如build-essential、gcc、g++和其他必要的开发工具: ``` sudo apt install build-essential gcc g++ ``` 4. 下载适用于您的CUDA版本的安装包。您可以从NVIDIA开发者网站上下载适用于Linux的CUDA安装包。 5. 在终端导航到下载的安装包所在的目录,并运行以下命令来使安装包可执行: ``` chmod +x cuda_installer_filename.run ``` (请将“cuda_installer_filename.run”替换为您下载的安装包的实际文件名) 6. 运行CUDA安装程序并按照提示进行安装: ``` sudo ./cuda_installer_filename.run ``` 7. 在安装过程,您将被要求接受许可协议、选择安装位置以及是否安装NVIDIA驱动程序等选项。根据您的需求进行选择。 8. 安装完成后,您需要配置环境变量。打开终端并编辑您的.bashrc文件: ``` nano ~/.bashrc ``` 9. 在文件的末尾添加以下行,并保存退出: ``` export PATH=/usr/local/cuda/bin:$PATH export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH ``` 10. 使用以下命令使更改生效: ``` source ~/.bashrc ``` 11. 最后,您可以验证CUDA是否正确安装和配置。在终端运行以下命令以检查CUDA版本: ``` nvcc --version ``` 如果安装成功,将显示CUDA版本号。 这样,您就可以在Ubuntu虚拟环境成功安装CUDA了。希望对您有所帮助!
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值