Ubuntu20.04 + GTX1050 + CUDA11.0 + cudnn8.0
置顶 • 1个月前 • 89次点击 • 来自 其他
标签: LinuxTensorFlow
记录下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