首页
学习
活动
专区
工具
TVP
发布
精选内容/技术社群/优惠产品,尽在小程序
立即前往
您找到你想要的搜索结果了吗?
是的
没有找到

linux服务器,卸载tensorflow CPU 安装PGU版

之前用的和学习的都是pytorch框架,现在要运行一个keras的代码,得安装tensorflow和keras,按一个教程,直接在pycharm里setting,点那个+很快就装好了tensorflow和keras,运行了几次发现运行特别慢,用nvidia-smi查看,发现根本没有用pgu跑,一番查找,最后发现安装的tensorflow本身是按CPU跑的,要用GPU跑,得安装tensorflow-gpu。 以下主要参考了https://blog.csdn.net/qq_38502918/article/details/108009692进行操作,成功安装了tensorflow-gpu版本的。 记录以下安装过程。 重点: CUDA的版本要与tensorflow-gpu的一定要对应,否则会出错。 注意点: 安装好tensorflow-gpu后,安装对应版本的keras版本。 https://blog.csdn.net/weixin_40109345/article/details/106730050

03
领券