ubuntu20.04安装tensorflow-gpu版本避坑

2021-03-07  本文已影响0人  都灵的夏天_

安装驱动

  1. 打开软件与更新


  2. 查看驱动



    切忌最后一个开源驱动不要选择(其他按版本关系来)
    3.卸载原有的开源驱动(如果你打开附加驱动,显示的最后一个,推荐操作)

#打开黑名单:
sudo gedit /etc/modprobe.d/blacklist-nouveau.conf
#添加两行语句:
blacklist nouveau
options nouveau modeset=0
#更新initramfs
sudo update-initramfs -u
#重启
reboot
#验证,终端输入语句,
lsmod | grep nouveau
#如果没有结果表示禁用成功。
  1. TensorFlow版本
    网上有很多要求下载指定版本的CUDA toolkit、 cuDNN ,根据的 不同TensorFlow 对应的版本不同
    tnsorflow版本,CUDA,cuDNN经过测试的构建配置
conda install tensorflow-gpu==1.15 
  1. 查看驱动版本
    由上图可见我的版本是**nvdia-driver-460 **版号就是460 这个很重要
    CUDA版本的选择:它依赖于显卡的驱动程序版本,如下表,参考官方文档。与下表进行对照选择兼容的CUDA版本。
    cuda与显卡驱动版本依赖
  2. 选择合适版本的驱动
conda list
2021-03-07 21-02-24 的屏幕截图.png

运行速度快很多了,虽然显卡很一般但也比原来好多了。希望大家也能一次成功。

附录

#静态显示GPU信息
 nvidia-smi
#配合watch命令, 让一秒刷新一次
 watch -n 1 nvidia-smi   

#输入
conda install -c conda-forge ipdb
#如果不行就用
pip install ipdb
上一篇下一篇

猜你喜欢

热点阅读