不同conda环境设置不同的cuda版本
1.建立激活环境下的文件夹，脚本。2.建立退出环境下的文件夹，脚本。
·
1.建立激活环境下的文件夹，脚本
mkdir -p /home/用户名/anaconda3/envs/环境名/etc/conda/activate.d
nano /home/用户名/anaconda3/envs/环境名/etc/conda/activate.d/activate.sh写入
#!/bin/sh
export PATH=/usr/local/<cuda版本>/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/<cuda版本>/lib64:$LD_LIBRARY_PATH
2.建立退出环境下的文件夹，脚本
mkdir -p /home/用户名/anaconda3/envs/环境名/etc/conda/deactivate.d
nano /home/用户名/anaconda3/envs/环境名/etc/conda/deactivate.d/deactivate.sh写入
#!/bin/sh
unset PATH
unset LD_LIBRARY_PATH
source ~/.bashrc
欢迎来到由智源人工智能研究院发起的Triton中文社区，这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂，为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
3
0- 0
已为社区贡献1条内容
所有评论(0)