Ubuntu20.04 + GTX1050 + CUDA11.0 + cudnn8.0

置顶 6个月前 704次点击 来自 其他

标签: Linux

记录下Ubuntu20.04 + GTX1050 + CUDA11.0 + cudnn8.0安装过程。

1. 安装GTX1050显卡驱动

Ubuntu20.04下提供了Additional Drivers用于安装驱动,很简单,我选择的驱动是 nvidia-driver-450

2.下载对应版本CUDA和cudnn

cuda_11.0.2_450.51.05_linux.run:
https://developer.download.nvidia.cn/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
cudnn-11.0-linux-x64-v8.0.5.39.tgz:
https://developer.nvidia.com/compute/machine-learning/cudnn/secure/8.0.5/11.0_20201106/cudnn-11.0-linux-x64-v8.0.5.39.tgz

安装cuda_11.0.2_450.51.05_linux.run:

sudo sh cuda_11.0.2_450.51.05_linux.run

配置环境变量

 vi ~/.bashrc
export PATH=/usr/local/cuda-11.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}

更新环境变量配置

 source ~/.bashrc

nvcc --version查看当前使用的cuda版本

sudo rm -rf /usr/local/cuda  #删除之前创建的软链接
sudo ln -s /usr/local/cuda-11.0  /usr/local/cuda  #创建新的软链接
nvcc --version

解压并复制文件到 /usr/local/cuda ,并添加读取权限

tar -xvzf cudnn-11.0-linux-x64-v8.0.5.39.tgz
cd cuda #进入解压后的cuda文件夹
sudo cp include/cudnn.h /usr/local/cuda/include/
sudo cp lib64/lib* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*

测试cudnn是否安装好了,运行一个sample即可

cd /usr/local/cuda/samples/1_Utilities/deviceQuery 
sudo make
./deviceQuery 

编译并测试带宽 bandwidthTest

cd ../bandwidthTest
sudo make 
./bandwidthTest
Card image cap
开发者雷

尘世间一个小小的开发者,每天增加一些无聊的知识,就不会无聊了

要加油~~~

技术文档 >> 系列应用 >>
热推应用
Let'sLearnSwift
学习Swift的入门教程
PyPie
Python is as good as Pie
标签