Ubuntu临时切换Cuda版本

最新推荐文章于 2024-05-06 16:20:47 发布

CExploer

最新推荐文章于 2024-05-06 16:20:47 发布

阅读量1.1k

点赞数

分类专栏： Ubuntu 服务器 Cuda 文章标签： ubuntu python linux

本文链接：https://blog.csdn.net/Exploer_TRY/article/details/128066877

版权

服务器同时被 3 个专栏收录

4 篇文章 0 订阅

订阅专栏

Ubuntu

3 篇文章 0 订阅

订阅专栏

Cuda

1 篇文章 0 订阅

订阅专栏

Ubuntu临时切换Cuda版本

Ubuntu临时切换cuda版本，也就是只是当前程序运行需要更换cuda版本，其他不变。

这个方法有个前提是你已经安装了各种cuda的版本，一般他们都会在/usr/local/目录下，并且都附有自己的cuda版本号。

此时更换新版本并不需要软连接到/usr/local/cuda。我们只需要在ubuntu写入如下命令。例如我们目前想临时用Cuda11.1训练程序：

export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64
export PATH=/usr/local/cuda-11.1/bin
export CUDA_HOME=/usr/local/cuda-11.1

export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python

这样即可。

nvcc -V

输出：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:01_CDT_2018
Cuda compilation tools, release 10.0, V10.0.130

继续输入命令：

export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64
export PATH=/usr/local/cuda-11.1/bin
export CUDA_HOME=/usr/local/cuda-11.1

export PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python

nvcc -V

新的输出：

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2020 NVIDIA Corporation
Built on Tue_Sep_15_19:10:02_PDT_2020
Cuda compilation tools, release 11.1, V11.1.74
Build cuda_11.1.TC455_06.29069683_0