[心得] Win10 TensorFlow 2.0 GPU 升级中的坑

楼主: brad0315 (老默)   2019-12-11 15:02:23
其实一开始只是想要升级Spyder到4.0的版本,这样就有内建暗黑主题可以用了。
但所有更新都是灾难的开始......
废话不多说,直接上我最后解决一切问题的步骤:
Step 1、打开Anaconda Prompt 进到你想更改的环境里,懒人如我没有建虚拟环境。
Step 2、更新conda :conda update conda
Step 3、更新Anaconda :conda update anaconda
Step 4、更新Spyder :conda update spyder
基本上到这边,Spyder就会更新成新版有内建暗黑主题的,但是可能会发现自己的
TensorFlow变成跑CPU版,速度差了十几倍。
Step 5、打开命令提示字符(cmd)
Step 6、安装TensorFlow2.0:pip install tensorflow-gpu
Step 7、安装 CUDA以及cuDNN: conda install cudnn=7.6 cudatoolkit=10.0
一定要这个版本,因为目前TensorFlow2.0还不支援最新CUDA10.1。
现在居然可以直接用conda安装,真的太感动,以前还要去官网载。
Step 8、修改gast版本:pip install gast==0.2.2
因为gast最新版本有Bug,所以请安装这个特定版本,不然会有Error,虽然可以无视。
也写成网志了,里面有提供简单的测试程式码
有兴趣的可以去看一下,无广告、音乐:
https://mortis.tech/2019/12/tutorial/921/
作者: roccqqck (ccqq)   2019-12-11 16:02:00
创虚拟比较简单 不用降其他东西版本conda create -n tfenv python=3.7 tensorflow-gpu=2.0create时也可以后面加个anaconda 把整包一起装到虚拟环境以后更新爆掉了话 也可以整个砍掉重来
作者: wargods8402 (维尼X维尼)   2019-12-11 16:12:00
借问 创虚拟环境不会很混乱吗...?
作者: roccqqck (ccqq)   2019-12-11 16:43:00
base才混乱吧旧project 套件版本更新后就error 或套件之间不相容 不用虚拟环境超容易烂掉的
作者: TuCH (谬客)   2019-12-11 16:48:00
学会创个虚拟环境可以减少很多灾难XDDD
作者: roccqqck (ccqq)   2019-12-11 16:50:00
pip install有时候会安装完才跟你说不相容 超可怕
楼主: brad0315 (老默)   2019-12-11 17:51:00
经过这次有烦到,会乖乖装虚拟QQ
作者: lionell   2019-12-11 20:40:00
官方版的话,github上fo20225有编好的whl不过都是搭配最新的cuda跟cudnn,可以参考看看
作者: s860134 (s860134)   2019-12-12 01:34:00
可以试试 pipenv
作者: yoyololicon (萝莉大好)   2019-12-14 12:51:00
用anaconda不创虚拟环境 excuse me WTF
作者: ruokcnn (Dean)   2019-12-18 15:42:00
看到楼上笑死

Links booklink

Contact Us: admin [ a t ] ucptt.com