疫情在家开始学习tensorflow,但是上手就碰到了一些问题:
环境:windows 10; tensorflow 2.1.0 (装的是Gpu版本);Cuda 10.2
GPU是 Nvidia的2060
开始做很简单的前向传播问题,发现报错"failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED"
如下图

然后百度了一下,根据提示加了这样一行 os.environ["CUDA_VISIBLE_DEVICES"] = "1"
加上后能跑起来了,但是还有一个错误提示 failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected
如下图:

根据我的理解,那就是GPU根本没有跑起来,上面只是用CPU跑了一下。我看了一下任务管理器,GPU占用一直只有2%,所以GPU根本没用上。再看GPU的编号是0,那我就把添加的那行修改成:os.environ["CUDA_VISIBLE_DEVICES"] = "0"
此时报错,和未添加这行的时候是一样的

但是我在意到一点,GPU占用有很短暂的一瞬是提高了的

后面应该还是需要GPU的,不能总用CPU跑着吧
所以来请教大佬如何解决这个问题?
环境:windows 10; tensorflow 2.1.0 (装的是Gpu版本);Cuda 10.2
GPU是 Nvidia的2060
开始做很简单的前向传播问题,发现报错"failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED"
如下图
然后百度了一下,根据提示加了这样一行 os.environ["CUDA_VISIBLE_DEVICES"] = "1"
加上后能跑起来了,但是还有一个错误提示 failed call to cuInit: CUDA_ERROR_NO_DEVICE: no CUDA-capable device is detected
如下图:
根据我的理解,那就是GPU根本没有跑起来,上面只是用CPU跑了一下。我看了一下任务管理器,GPU占用一直只有2%,所以GPU根本没用上。再看GPU的编号是0,那我就把添加的那行修改成:os.environ["CUDA_VISIBLE_DEVICES"] = "0"
此时报错,和未添加这行的时候是一样的
但是我在意到一点,GPU占用有很短暂的一瞬是提高了的

后面应该还是需要GPU的,不能总用CPU跑着吧
所以来请教大佬如何解决这个问题?
