关于tensorflow无法使用gpu
2024-10-15 20:39:32
python3.6 无法使用tensorflow gpu
环境名称 test1
在控制台里进入环境
conda activate test1
使用python
python
查看gpu能否使用
print(tf.config.list_physical_devices('GPU'))
cuda版本查看 nvcc -V
10.2.89
判断是否使用了gpu:
推荐的方法是检查TensorFlow是否使用GPU如下:
tf.config.list_physical_devices('GPU')
输出:【这个很重要】
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
以下内容还将返回您的GPU设备的名称。
import tensorflow as tf
tf.test.gpu_device_name()
如果non-GPU
安装了该软件包的版本,则该函数还将返回False
。使用tf.test.is_built_with_cuda
来验证是否TensorFlow是建立在CUDA的支持。
*注意*:tf.test.is_gpu_available
已弃用。请参考这里
警告:此功能已被弃用。它将在将来的版本中删除。更新说明:改用tf.config.list_physical_devices('GPU')。
我最后终于搞出来了
用的就是CUDA_VISIBLE_DEVICES网上的方法
最新文章
- 【Swift】Alamofile网络请求数据更新TableView的坑
- IOS-WebViewJavascriptBridge使用说明
- win7远程桌面连接windows2008服务器,连接启动很慢。我是这样解决的。
- linux系统limit知识
- 基于Metronic的Bootstrap开发框架经验总结(9)--实现Web页面内容的打印预览和保存操作
- 【mysql】数据库使用的一些规范
- August 28th 2016 Week 36th Sunday
- 簡單SQL存儲過程實例
- Arch 常用工具
- 读UNDO引发的db file sequential read
- 懒猫们终究要付出代码(本领是一生的),鲸鱼们的短视(逐小利而暴死)——这么说我应该只去互联网公司:IM,云存储,邮箱(别的一概不考虑)
- Codeforces 433 C. Ryouko&;#39;s Memory Note
- 命令更新emacs
- curl 模拟表单post文件
- 剑指Offer 3. 从尾到头打印链表 (链表)
- mysql主从备份功能配置与測试
- mysql系统变量查询
- 【Coursera】SecondWeek(1)
- matlab彩色图像插值
- 配置kernel的log buf大小(如果kmsg log被覆盖)