作者brad0315 (Mortis)
看板Python
标题[心得] Win10 TensorFlow 2.0 GPU 升级中的坑
时间Wed Dec 11 15:02:23 2019
其实一开始只是想要升级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/
--
※ 发信站: 批踢踢实业坊(ptt.cc), 来自: 140.113.76.101 (台湾)
※ 文章网址: https://webptt.com/cn.aspx?n=bbs/Python/M.1576047746.A.BB7.html
1F:→ roccqqck: 创虚拟比较简单 不用降其他东西版本 12/11 16:02
2F:→ roccqqck: conda create -n tfenv python=3.7 tensorflow-gpu=2.0 12/11 16:03
3F:→ roccqqck: create时也可以後面加个anaconda 把整包一起装到虚拟环 12/11 16:06
4F:→ roccqqck: 境 12/11 16:06
5F:推 roccqqck: 以後更新爆掉了话 也可以整个砍掉重来 12/11 16:10
6F:推 wargods8402: 借问 创虚拟环境不会很混乱吗...? 12/11 16:12
7F:→ roccqqck: base才混乱吧 12/11 16:43
8F:→ roccqqck: 旧project 套件版本更新後就error 或套件之间不相容 不 12/11 16:45
9F:→ roccqqck: 用虚拟环境超容易烂掉的 12/11 16:45
10F:推 TuCH: 学会创个虚拟环境可以减少很多灾难XDDD 12/11 16:48
11F:→ roccqqck: pip install有时候会安装完才跟你说不相容 超可怕 12/11 16:50
12F:→ brad0315: 经过这次有烦到,会乖乖装虚拟QQ 12/11 17:51
13F:→ lionell: 官方版的话,github上fo20225有编好的whl 12/11 20:40
14F:→ lionell: 不过都是搭配最新的cuda跟cudnn,可以参考看看 12/11 20:41
15F:推 s860134: 可以试试 pipenv 12/12 01:34
16F:嘘 yoyololicon: 用anaconda不创虚拟环境 excuse me WTF 12/14 12:51
17F:推 ruokcnn: 看到楼上笑死 12/18 15:42