tensorflow不能用显卡跑程序的问题

xiaoxiao2021-08-25  376

刚入坑机器学习。跑了四五天了,一遍还没跑完。

查进程号

ps -u  uesrid(自己的用户名)

 nvidia-smi 查看 显卡状态。发现没有自己的pid  进程号。

然后发现程序里CUDA_VISIBLE_DEVICES =1 python my.py  依旧没有帮助。

后来发现自己安装的tensorflow 版本有问题。卸载之后安装tensorflow-gpu版本 一切正常。

conda uninstall tensorflow -y

pip install tensorflow-gpu

前后速度简直是从挣钱到花钱

 

转载请注明原文地址: https://www.6miu.com/read-4823547.html

最新回复(0)