![](https://img-blog.csdnimg.cn/20201014180756757.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow2.0
白色小靴
学生
展开
-
tensorflow2.0/2.1-gpu版本安装教程
本人电脑配置RTX2070,支持CUDA10 下载gpu版本要注意cuda的版本是否与电脑的相匹配,至于能匹配什么版本请自行百度 但是要注意!tf2.0只能与CUDA10进行匹配,所以不支持的小伙伴就可以不用继续往下看了:) 感谢本链接https://blog.csdn.net/wangbowj123/article/details/89381562,其中有些地方详细说明了一下。有些图直接转载了。...原创 2020-03-03 18:00:40 · 2671 阅读 · 5 评论 -
tensorflow2.0显存自适应分配
gpu按需使用 gpus = tf.config.experimental.list_physical_devices(device_type='GPU') for gpu in gpus: tf.config.experimental.set_memory_growth(gpu, True)原创 2019-11-06 19:59:04 · 6049 阅读 · 4 评论