自学内容网 自学内容网

linux环境安装cuda toolkit

1 全新安装
如果环境中没安装过cuda版本, 这种情况下比较简单。 直接在https://developer.nvidia.com/cuda-toolkit-archive选择对应版本下载安装即可。

如下为安装cuda toolkit 11.8.
在这里插入图片描述

2 环境中已经存在其他版本
这种情况下比较复杂一些。 首先要确认最高支持的版本,通过nvidia-smi查看,

这里显示cuda版本是11.8, 这里显示的是驱动的版本。由于驱动是向下兼容的,因此只要cuda toolkit版本小于等于这个即可。

安装时为了避免更新驱动导致其他问题, 在执行sudo sh cuda_11.8.0_520.61.05_linux.run安装过程中, 我们把驱动这一项去掉, 只安装cuda toolkit.

**

安装nvidia-cuda-toolkit

**

wget https://developer.download.nvidia.com/compute/cuda/12.4.0/local_installers/cuda_12.4.0_550.54.14_linux.run
sh cuda_12.4.0_550.54.14_linux.run

在这里插入图片描述
安装完成后提示信息如下:
在这里插入图片描述
由于安装时没有删除掉旧版本 ,因此现在环境中应该会存在多个cuda toolkit版本, 并且默认还是旧版本, 可以通过nvcc -V命令查看。
在这里插入图片描述
多版本切换
设置PATH和LD_LIBRARY_PATH两个环境变量, 把需要用到的版本路径加到最前面, 那么就会优先用到对应的版本。

 export PATH=/usr/local/cuda-11.8/bin:$PATH
 export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH

设置好之后再通过nvcc -V查看, 发现切换到新版本了:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2022 NVIDIA Corporation
Built on Wed_Sep_21_10:33:58_PDT_2022
Cuda compilation tools, release 11.8, V11.8.89
Build cuda_11.8.r11.8/compiler.31833905_0


原文地址:https://blog.csdn.net/huiyayaya/article/details/143859804

免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!