最近看了一些博文安装cuda,cudnn,总结了一下,流程如下:
一、显卡驱动的安装
1.首先是显卡驱动,需要安装推荐版本的显卡驱动,使用命令ubuntu-drivers devices;
2.安装驱动,使用命令sudo apt install nvidia-版本号;
3.安装完成后使用命令nvidia-smi,查看是否安装成功;
二、安装cuda
1.进cuda官网下载和驱动适配的cuda版本,下载对应的.run文件;选择版本的参考链接如下:Ubuntu 18.04安装Tensorflow(GPU)_Linux教程_Linux公社-Linux系统门户网站
2.安装cuda,使用命令sudo sh cuda文件名;
3.安装完成后,可使用cat /usr/local/cuda/version.txt,查看cuda的版本号,如果安装成功就会显示cuda的版本号,否则安装失败;
4.注:在安装cuda时,一直yes或者回车,如果提示是否安装显卡驱动,选择no,不要用这个显卡驱动(一直按回车键安装,碰到是否安装Driver可以选择否(因为之前已经安装),Samples也可以不安装)
三、安装cudnn
1.进入cudnn官网,需要注册登录,才能下载cudnn,下载相应版本的cudnn for linux;选择版本的参考链接如下:Ubuntu 18.04安装Tensorflow(GPU)_Linux教程_Linux公社-Linux系统门户网站
2.下载完成后,进入相应的目录使用命令tar -zxvf 文件名 进行解压缩;
3.复制cudnn中相关文件到cuda中,把相应文件复制到/usr/local/cuda/ CUDA安装文件夹里
# 解压
tar -zxvf cudnn-9.0-linux-x64-v7.tgz
# 复制相应文件
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64/
sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include/
# 所有用户可读
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
四、安装anaconda
下载anaconda,然后使用命令行进行安装
五、安装tensorflow-gpu
在anaconda环境下安装tensorflow-gpu版本,使用命令conda install tensorflow-gpu进行tensorflow的安装
六、安装pytorch
在anaconda环境下安装pytorch,直接 conda install pytorch torchvision cudatoolkit=9.0 -c pytorch,这个命令的格式是这样的,但是具体的参数什么的需要进pytorch的官网进行查看。
到此,全部的环境搭建以及安装就完成了。。。