1.~/.bashrc 文件是 Bash shell 的配置文件之一, 主要用于设置用户特定的环境变量、别名、函数以及其他 shell 配置。可在该文件中进行cuda环境变量的添加,可以使用vi编辑器对该文件进行编辑,按 i 键进入插入模式。

vi ~/.bashrc

添加或修改 CUDA 相关的环境变量,可以找到合适的位置(通常是文件末尾),然后添加以下内容:

export CUDA_HOME=/home/kwatb/CUDA11.1.1
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=/home/kwatb/CUDA11.1.1/lib64
export PATH=$LD_LIBRARY_PATH:$PATH
保存并退出
  • 按 Esc 键返回到普通模式。
  • 输入 :wq 并按 Enter 键保存更改并退出 vi 编辑器。
    • :w 表示写入(保存)。
    • :q 表示退出。

如果你不想保存更改并退出,可以输入 :q! 并按 Enter 键。

使更改生效

在保存并退出 vi 编辑器后,你需要使更改立即生效。你可以通过以下命令来实现:

source ~/.bashrc

验证更改

echo $CUDA_HOME
echo $PATH
echo $LD_LIBRARY_PATH

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐