tensorflow
关注
分享
扫一扫
Mr.NingYL
这个作者很懒，什么都没留下…
展开
专栏收录文章
- 默认排序
- 最新发布
- 最早发布
- 最多阅读
- 最少阅读
-
tensorflow2 设置显存按需增长
# 获取所有 GPU 设备列表gpus = tf.config.experimental.list_physical_devices('GPU')if gpus: try: # 设置 GPU 显存占用为按需分配，增长式 for gpu in gpus: tf.config.experimental.set_memory_growth(gpu, True)except RuntimeError as e: # 异常处理 print(e)...原创 2021-03-19 20:01:14 · 691 阅读 · 0 评论 -
解决tensorflow-gpu2.0与CUDA10.1的兼容问题 dlerror: libcudart.so.10.0: cannot open shared object file
最近要给一台共用的ubuntu18.04配置tf2.0 gpu环境，原机器上是tf1.x版本+CUDA10.1，gpu可以正常使用和运行，但是虚拟环境安装tf2.0-gpu后，执行tf.test.is_gpu_available()出现了tensorflow/stream_executor/platform/default/dso_loader.cc:55] Could not load dy...原创 2019-12-05 19:33:15 · 7797 阅读 · 2 评论