GPU机器无法使用GPU

对于GPU机器无法使用GPU,可能有以下原因:

1、没有相应的cuda环境,即cuda和cudnn

2、有安装cuda但是没有添加到系统环境变量,添加环境变量即可,例如linux下以cuda10.1为例

export PATH=$PATH:/usr/local/cuda-10.1/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-10.1/lib64

3、没有安装GPU版本的框架,例如tensorflow-gpu

例如使用pip install tensorflow=2.0之后安装可能只是CPU版本的tensorflow,执行查看是否能调用gpu

import tensorflow as tf
print(tf.test.is_gpu_available())

执行后,结果返回结果是false。此时执行

pip install tensorflow-gpu=2.0

可以发现版本升级为gpu的:

The following packages will be DOWNGRADED:
  _tflow_select                                   2.3.0-mkl --> 2.1.0-gpu
  tensorflow                       2.0.0-mkl_py36hef7ec59_0 --> 2.0.0-gpu_py36h6b29c10_0
  tensorflow-base                  2.0.0-mkl_py36h9204916_0 --> 2.0.0-gpu_py36h0ec5d1f_0

 

  • 0
    点赞
  • 5
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值