TensorFlow学习笔记 速记2 报错:failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_INVALID_DEVICE
2024-10-19 11:49:06
版本:
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
最新文章
- GJM : C#语言学习笔记
- 【Java EE 学习 47】【Hibernate学习第四天】【sesion】【一级缓存】【懒加载】
- Git过滤文件和文夹
- 计算机启动boot
- SQLserver2012 修改数据库架构
- Metasploit启动
- 深入理解C#:编程技巧总结(一)
- BZOJ 3572 世界树
- eclipse下使用hibernate tools实现hibernate逆向工程
- .Net类型与JSON的映射关系
- C++_基础_运算符重载2
- JS运行机制之 Event Loop 的思考
- 使用JS+Three.js+Echart开发商场室内地图客流信息统计功能
- Linux之部署前后端分离项目
- pandas常用函数之shift
- vmlinux,zImage,bzImage,vmlinuz,uImage,关系
- Bukkit编程之动态向yml文件中添加属性
- Java反序列化修复方案
- vscode vue 格式化 和emmet 提示
- where_1
热门文章
- Xamarin.Forms使用Slider注意问题
- [Contest20180313]灵大会议
- JS面向对象之作用域
- 如何垂直居中元素(浮动元素&;居中一个<;img>;)?
- webpack配置构建环境问题汇总
- 北京极科极客科技有限公司 http://www.hiwifi.com/
- nagios系列教程地址
- appium自动化,失败自动截图
- virtualbox虚拟机ubuntu操作系统,设置网络互通、访问,能访问虚拟机swoole的http服务
- netty4 ServerBootstrap.bind(port) debug