YouTube:https://www.youtube.com/c/xugaoxiang;Bilibili: 迷途小书童的Note;微信公众号: Dev_Club

ubuntu安装CUDA

人工智能 迷途小书童 0评论

软硬件环境

  • ubuntu 18.04 64bit
  • NVidia GTX 1070
  • CUDA 10.1
  • cudnn 7.6.3

视频看这里

安装nvidia驱动

进入没有安装驱动的ubuntu系统,屏幕显示是不正常的,分辨率偏小。从官网下载RUN文件, 比如我这里的NVIDIA-Linux-x86_64-435.21.run,这里尽量选择较新的版本

sudo apt install build-essential 
chmod a+x NVIDIA-Linux-x86_64-435.21.run
sudo ./NVIDIA-Linux-x86_64-435.21.run

安装过程基本上就是选中yes或者no,装完后重启系统,就会发现屏幕显示正常了,分辨率也变高了。这时候,我们可以通过nvidia提供的一些命令来查看系统中显卡的信息。如非常有用的nvidia-smi

安装CUDA

接下来安装cuda

sudo ./cuda_10.1.243_418.87.00_linux.run

然后配置下,编辑~/.bashrc,添加相应的环境变量

export PATH=/usr/local/cuda-10.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH

最后

source ~/.bashrc

这样几个重要的环境变量就立马生效了。

安装cudnn

到官网https://developer.nvidia.com/cudnn去下载,这里需要注册,有点麻烦

tar xvf cudnn-10.1-linux-x64-v7.6.3.36.tgz
cd cuda
sudo cp cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp -a cuda/lib64/libcudnn*  /usr/local/cuda-10.1/lib64
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*

测试

安装完毕,我们来测试下环境是否都ok

刚好我这里跑了示例,所以你能看到gpu基本上是满负荷工作

参考资料

喜欢 (0)
发表我的评论
取消评论

表情

Hi,您需要填写昵称和邮箱!

  • 昵称 (必填)
  • 邮箱 (必填)
  • 网址