ubuntu16.04+cuda9.2+tensorflow-gpu1.8.0+torch1.3.0 问题记录

遇到的问题

  • 1.显卡驱动显示正常,但是pytorch和tensorflow-gpu都显示无法寻找到显卡设备
    • 问题:
    • 问题解决:

1.显卡驱动显示正常,但是pytorch和tensorflow-gpu都显示无法寻找到显卡设备

问题:

nvidia-smi # 显示正常
lspci | grep vga # 显示正常

ubuntu16.04+cuda9.2+tensorflow-gpu1.8.0+torch1.3.0 问题记录_第1张图片但是,pytorch和tensorflow寻找不到设备,即torch.cuda.isavaliable()显示false
ubuntu16.04+cuda9.2+tensorflow-gpu1.8.0+torch1.3.0 问题记录_第2张图片

问题解决:

命令

ubuntu-drivers devices

ubuntu16.04+cuda9.2+tensorflow-gpu1.8.0+torch1.3.0 问题记录_第3张图片看了一下自己的显卡需求,虽然384也在适配驱动中,但是最后仔细查看torch和tensorflow版本之后发现没错,就确定是显卡驱动的问题或者是显卡本身的问题,就重装了显卡驱动。
手动下载了430驱动 Nvidia_linux_x86_64_430.50.run,按照教程重装了一遍,测试可用了。

你可能感兴趣的:(环境配置)