感觉tensorfloe-gpu限制比较多
探索出来的一种可行办法:
1.安装miniconda(anaconda不行,如果本机装了的话,建议卸载)
2.在base环境下安装conda install tensorflow=2.0.0
这里我试过最新的2.3版本,不太行。2.0.0版本会附带装上cudatoolkit(10.0.130)和cudnn(7.6.5)
比较简洁
然后在vscode下选择对应的ipython kernel
3.测试gpu版本是否可用
import tensorflow as tf
if tf.test.gpu_device_name():
print('Default GPU Device: {}'.format(tf.test.gpu_device_name()))
else:
print("Please install GPU version of TF")
出现上述结果,则证明可用