在运行tensorflow代码时,在相同的代码在jupyter中可以运行,但在pycharm中报错:failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILE
原因是没有指定gpu,在开头加入以下代码即可:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"#指定gpu编号,在任务管理器的性能中可以看到
关于tensorflow显存占用更详细的介绍:链接: link.
在运行tensorflow代码时,在相同的代码在jupyter中可以运行,但在pycharm中报错:failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILE
原因是没有指定gpu,在开头加入以下代码即可:
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"#指定gpu编号,在任务管理器的性能中可以看到
关于tensorflow显存占用更详细的介绍:链接: link.