linux系统下切换不同CUDA针对单用户

原创于 2024-11-26 11:17:09 发布 · 254 阅读

CC 4.0 BY-SA版权

文章标签：

编辑你的 shell 配置文件（例如 .bashrc 或 .zshrc）：

nano ~/.bashrc

在文件末尾添加或修改 CUDA 路径：

# CUDA 11.7
export PATH=/usr/local/cuda-11.7/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.7/lib64:$LD_LIBRARY_PATH

# CUDA 12.1
# export PATH=/usr/local/cuda-12.1/bin:$PATH
# export LD_LIBRARY_PATH=/usr/local/cuda-12.1/lib64:$LD_LIBRARY_PATH

保存并退出编辑器（在 nano 中按 Ctrl + X，然后按 Y，再按 Enter）。
source ~/.bashrc

以上两步很重要。

最后就是 nvcc -V检查了。