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的处理(代