感觉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")

 

出现上述结果,则证明可用

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐