1.cuda 11.8网址

CUDA Toolkit 11.8 Downloads | NVIDIA Developer

2.安装

sh cuda_11.8.0_520.61.05_linux.run 

直接accept

移动上下到driver上,再enter一下取消driver的更新

更新一下.bashrc,cp -r ~/.bashrc  xxx

source ~/.bashrc

export PATH=/usr/local/cuda-11.8/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.8/lib64:$LD_LIBRARY_PATH
export PATH=/home/miniconda3/bin:$PATH
. /home/miniconda3/etc/profile.d/conda.sh
conda activate

一般安装完路径就在/usr/local目录下,更新驱动需要重新启动reboot,只更新cuda不需要重新启动。

3.cudnn安装

地址:https://developer.nvidia.com/rdp/cudnn-download

tar -xf cudnn-linux-x86_64-8.9.4.25_cuda11-archive.tar.xz
cp -r cudnn.h /usr/local/cuda-11.8/include/
cp -r lib/libcudnn* /usr/local/cuda-11.8/lib64*

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐