python3.6 无法使用tensorflow gpu
环境名称 test1
在控制台里进入环境
conda activate test1
使用python
python
查看gpu能否使用
print(tf.config.list_physical_devices('GPU'))
cuda版本查看 nvcc -V
10.2.89
判断是否使用了gpu：
推荐的方法是检查TensorFlow是否使用GPU如下：
tf.config.list_physical_devices('GPU')
输出：【这个很重要】
[PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')]
以下内容还将返回您的GPU设备的名称。
import tensorflow as tf
tf.test.gpu_device_name()
如果non-GPU安装了该软件包的版本，则该函数还将返回False。使用tf.test.is_built_with_cuda来验证是否TensorFlow是建立在CUDA的支持。
*注意*：tf.test.is_gpu_available已弃用。请参考这里
警告：此功能已被弃用。它将在将来的版本中删除。更新说明：改用tf.config.list_physical_devices（'GPU'）。
我最后终于搞出来了
用的就是CUDA_VISIBLE_DEVICES网上的方法