双系统ubuntu18.04 安装cuda+cudnn

最近看了一些博文安装cuda,cudnn,总结了一下,流程如下:

一、显卡驱动的安装

二、安装cuda

三、安装cudnn

四、安装anaconda

五、安装tensorflow-gpu

六、安装pytorch


一、显卡驱动的安装

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的官网进行查看。

到此,全部的环境搭建以及安装就完成了。。。

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值