TensorFlow报错failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

训练模型时报错：
tensorflow/stream_executor/cuda/cuda_blas.cc:510] failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED
2019-04-10 12:57:35.640700: W

tensorflow/stream_executor/stream.cc:2130] attempting to perform BLAS operation using StreamExecutor without BLAS support

环境：
win10
TensorFlow -gpu-1.3 + cuda9.0
pycharm

解决方法：
在程序中加入
import os
os.environ[“CUDA_VISIBLE_DEVICES”] = “1”