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 · 600 阅读 · 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 · 7658 阅读 · 2 评论