Ubuntu18.04使用Anaconda安装Tensorflow-gpu

最新推荐文章于 2024-08-20 18:55:43 发布

广东省热心网友

最新推荐文章于 2024-08-20 18:55:43 发布

阅读量1.5w

点赞数 3

文章标签： ubuntu18.04 tensorflow anaconda NVIDIA cuda

本文链接：https://blog.csdn.net/joker_or/article/details/80333253

版权

本文详细记录了在Ubuntu18.04系统中，通过Anaconda安装Tensorflow-GPU的全过程，包括NVIDIA驱动的安装、CUDA9.0和cuDNN7.0的配置、Anaconda的安装以及Tensorflow环境的创建和验证。在安装过程中，强调了对驱动版本和CUDA版本的要求，以及环境变量的配置，最后在jupyter-notebook中成功运行tensorflow的验证步骤。

摘要由CSDN通过智能技术生成

经过了几天的环境搭建，踩了不少坑，将经验分享一下～

一、安装NVIDIA驱动

1.卸载原有NVIDIA驱动

#使用apt方法安装：  
sudo apt-get remove --purge nvidia*

#使用.run文件安装：
sudo chmod +x *.run  
sudo ./NVIDIA-Linux-x86_64-384.59.run --uninstall

2.禁用nouveau驱动

sudo gedit /etc/modprobe.d/blacklist.conf

在末尾添加:

blacklist nouveau

然后执行：sudo update-initramfs -u

重启后执行：lsmod | grep nouveau。如果没有屏幕输出，说明禁用nouveau成功。

3.命令行安装驱动

确定自己所需要安装的驱动版本：

ubuntu-drivers devices

可以看到有目前可以安装的驱动:

driver   : nvidia-340 - distro non-free
driver   : nvidia-driver-390 - distro non-free recommended
driver   : xserver-xorg-video-nouveau - distro free builtin