Ubuntu 18.04 + CUDA 9.0 + cuDNN + Tensorflow测试

本文详细介绍了如何在Ubuntu 18.04系统上,针对GeForce GTX 1050 Ti显卡,安装推荐的nvidia-driver-390驱动,降低gcc和g++版本以适应CUDA 9.0的要求,然后安装CUDA 9.0,并添加环境变量。接着,文章指导如何注册NVIDIA账号下载cuDNN v7.0.5并安装,最后通过Python测试Tensorflow的运行情况。
摘要由CSDN通过智能技术生成

显卡驱动

检测显卡型号及系统推荐显卡驱动

fh@fh-pc:~$ ubuntu-drivers devices
== /sys/devices/pci0000:00/0000:00:01.0/0000:01:00.0 ==
modalias : pci:v000010DEd00001C82sv000019DAsd00002463bc03sc00i00
vendor   : NVIDIA Corporation
model    : GP107 [GeForce GTX 1050 Ti]
manual_install: True
driver   : nvidia-driver-390 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin

显卡型号:GeForce GTX 1050 Ti

显卡驱动:nvidia-driver-390(系统推荐)

自动安装推荐的显卡驱动

fh@fh-pc:~$ sudo ubuntu-drivers autoinstall

手动选择显卡驱动版本

fh@fh-pc:~$ sudo apt install nvidia-390

gcc&g++降级

CUDA 9.0仅支持gcc&g++ 6.0及以下版本,而Ubuntu 18.04预装gcc&g++ 7.2,需要手动降级

sudo apt-get install gcc
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值