TensorFlow报错:failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE

Posted

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了TensorFlow报错:failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE相关的知识,希望对你有一定的参考价值。

版本:

tensorflow-gpu

原因:

在创建session时没有使用我想让它用的gpu

解决方案:

1. 在python程序中:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "0"

2. 运行时:

CUDA_VISIBLE_DEVICES=0 python **.py

3. 直接在环境变量中 export

export CUDA_VISIBLE_DEVICES=‘0‘
python **.py

  

以上是关于TensorFlow报错:failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE的主要内容,如果未能解决你的问题,请参考以下文章

启动Sonar报错,ERROR: [1] bootstrap checks failed [1]: system call filters failed to install

Windows下安装Tensorflow报错 “DLL load failed:找不到指定的模块"

解决import tensorflow报错:ImportError: DLL load failed: 找不到指定的模块

Python2.7 win7系统串口报错:raise SerialException('call to ClearCommError failed')

codis3.2安装报错dashboard.go:369: [PANIC] call rpc create-proxy to dashboard 127.0.0.1:18080 failed的处理(代

Tensorflow+keras解决 Fail to find the dnn implementation.