jupyter notebook使用tensorflow-gpu的问题记录
前提:
-
在虚拟环境tensorflow-gpu中已经使用conda install tensorflow-gpu==2.6.0
安装好了tensorflow-gpu(cuda和cudnn应该是随该命令一并安装的)。
-
已经将该环境添加到了jupyter notebook中
问题:
使用tf.test.is_gpu_available()
测试是否安装成功
-
终端中显示True
-
jupyter notebook 显示False
解决:
目前还不知道是什么问题,如果有大佬知道,还请不吝赐教。