赞
踩
https://developer.nvidia.com/cuda-toolkit-archivehttps://developer.nvidia.com/cuda-toolkit-archive
- wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda_11.7.0_515.43.04_linux.run
- sudo sh cuda_11.7.0_515.43.04_linux.run
- export PATH=/usr/local/cuda/bin:$PATH
- export LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH
https://developer.nvidia.com/rdp/cudnn-archive
cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz
tar -xf cudnn-linux-x86_64-8.9.2.26_cuda11-archive.tar.xz
然后,使用下面两条指令复制cuda文件夹下的文件到 /usr/local/cuda/lib64/ 和 /usr/local/cuda/include/ 中。
- cp cuda/lib64/* /usr/local/cuda-11.0/lib64/
- cp cuda/include/* /usr/local/cuda-11.0/include/
cat /usr/local/cuda-11.0/include/cudnn_version.h | grep CUDNN_MAJOR -A 2
至此,cudatoolkit 和 cudnn 都已安装完成,可以尽情利用GPU加速搞AI算法了
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。