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