1、cuda
https://developer.nvidia.com/cuda-toolkit-archive
选deb或runfile格式
下载完成后,命令行sudo sh cuda_version_linux.run(对应自己下载的文件名tab补齐),过程中选择是否安装驱动
设置环境变量,打开主目录下的 .bashrc文件添加如下路径
(对应自己版本修改)
export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/lcoal/cuda-10.1/lib64:$LD_LIBRARY_PATH"
终端运行:source ~/.bashrc
检查:nvcc --version
2、cudnn
选定版本下载
https://developer.nvidia.com/rdp/cudnn-archive
解压下载的文件,可以看到cuda文件夹,在当前目录打开终端,执行如下命令
sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*
3、查看版本
查看Nvidia显卡的详细信息 nvidia-smi
查看cuda 版本: cat /usr/local/cuda/version.txt or nvcc -V
查看cudnn版本: cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2