下载cuda12.1
官网网址
- wget https://developer.download.nvidia.com/compute/cuda/12.1.1/local_installers/cuda_12.1.1_530.30.02_linux.run
- sudo sh cuda_12.1.1_530.30.02_linux.run
复制代码 !import! 假如已经安装驱动,则不要选择dirver那项
添加环境变量
vim ~/.bashrc
- export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-12.1/lib6
- export PATH=$PATH:/usr/local/cuda-12.1/bin
- export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-12.1
复制代码 source ~/.bashrc
安装cudnn
参考教程
- wget https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.7/local_installers/12.x/cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz
- tar -xvf wget https://developer.nvidia.com/downloads/compute/cudnn/secure/8.9.7/local_installers/12.x/cudnn-linux-x86_64-8.9.7.29_cuda12-archive.tar.xz
- sudo cp cudnn-*-archive/include/cudnn*.h /usr/local/cuda-12.1/include
- sudo cp -P cudnn-*-archive/lib/libcudnn* /usr/local/cuda-12.1/lib64
- sudo chmod a+r /usr/local/cuda/include/cudnn*.h /usr/local/cuda/lib64/libcudnn*
复制代码 强烈建议参考 https://blog.csdn.net/qq_44961869/article/details/115954258
https://blog.csdn.net/qq_45754436/article/details/140323984
免责声明:如果侵犯了您的权益,请联系站长,我们会及时删除侵权内容,谢谢合作!更多信息从访问主页:qidao123.com:ToB企服之家,中国第一个企服评测及商务社交产业平台。 |