小白中的小白求助tensorflow使用

我已经全部安装好了，Jupyter，spyder，pycharm(cpu版和gpu版并存），请问pycharm里怎么调用-gpu呢，还是import tensorflow as tf 吗？（没装gpu版前是这个），我怎么知道是不是调用了gpu呢。谢谢

安装cuda和cudnn

需要另外安装cudnn和CUDA。
另外，tansorflow2.5版本以后，只需要pip tensorflow即可，在安装了cudnn和CUDA以后就可以调用N显卡了。装gpu版本反而有问题。

下载贴吧APP
看高清直播、视频！

分享到: