给虚拟环境指定cuda

最新推荐文章于 2025-03-10 12:30:31 发布

小Aer

最新推荐文章于 2025-03-10 12:30:31 发布

阅读量5.6k

点赞数 3

分类专栏：笔记文章标签： cuda anaconda linux

本文链接：https://blog.csdn.net/qq_41542989/article/details/118467309

版权

笔记专栏收录该内容

77 篇文章

订阅专栏

这篇博客记录了如何在conda虚拟环境中配置CUDA 10.0，以便运行依赖较低CUDA版本的代码。作者首先创建了环境激活和退出的脚本文件，分别设置和恢复CUDA_HOME及LD_LIBRARY_PATH环境变量，确保在激活环境后能指向CUDA 10.0，并在退出环境时恢复到系统原有的CUDA设置。通过这种方法，解决了CUDA版本不兼容的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

这个问题其实在zhihu看到过，但是现在没找到，就把按自己代码中的过程放在这里做个笔记

原因：

因为一些github代码比较老，导致torch版本比较老，只能支持cuda10.0，但由于自己电脑是10.1，所以有了在conda虚拟环境中配置cuda的想法（Tip：一般cuda版本向下兼容，如果自己电脑支持最高10.1，那么比10.1低的版本按道理都可以，兼容问题可查看我另外一篇文章link）

操作：

执行两个步骤（下面注意修改为自己的路径）：

增加启动文件：

#建立退出环境下的文件夹
mkdir -p /home/你自己的用户名/anaconda3/envs/你自己的环境名字/etc/conda/activate.d
#写入脚本
vi /home/你自己的用户名/anaconda3/envs/你自己的环境名字/etc/conda/activate.d/activate.sh

写入以下内容，我需要的是10.0

ORIGINAL_CUDA_HOME=$CUDA_HOME
ORIGINAL_LD_LIBRARY_PATH=$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda-10.0
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH

增加退出文件

#建立退出环境下的文件夹
mkdir -p /home/你自己的用户名/anaconda3/envs/你自己的环境名字/etc/conda/deactivate.d
#写入脚本
vi /home/你自己的用户名/anaconda3/envs/你自己的环境名字/etc/conda/deactivate.d/deactivate.sh

写入以下内容

export CUDA_HOME=$ORIGINAL_CUDA_HOME
export LD_LIBRARY_PATH=$ORIGINAL_LD_LIBRARY_PATH
unset ORIGINAL_CUDA_HOME
unset ORIGINAL_LD_LIBRARY_PATH

测试

首先先在本机查看环境变量 CUDA_HOME

echo $CUDA_HOME

输出为空，表示回到自己机器环境

激活环境后

echo $CUDA_HOME

结果为/usr/local/cuda-10.0
大功告成。