![](https://img-blog.csdnimg.cn/20201014180756916.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
python
尤利乌斯.X
这个作者很懒,什么都没留下…
展开
-
tensorflow在AMD上用ROCm框架实现GPU训练(亲测有效)
众所周知,a卡对于tensorflow gpu的支持很小,要想使用的话一般都得配n卡的电脑。而我只是感兴趣一个项目,想要跑出来一个模型,如果不用gpu加速,可能得跑一个月,所以不得不寻找在a卡上的解决方案。好在,终于找到了一个解决方案,虽然只是取巧,速度也不如正常n卡跑的快,但已经很不错了。我的配置:显卡:RX 550所用系统:虚拟机ubuntu 18注:如果是虚拟机的话,最好分配3GB内...原创 2019-08-17 19:50:38 · 20839 阅读 · 20 评论 -
win10完整Tensorflow-GPU环境搭建教程-附CUDA+cuDNN安装过程
最近一直想要用GPU训练一个五子棋AI,无奈自己的笔记本是A卡,速度太慢,计算一局需要7、8分钟。所以在之前安了Ubuntu虚拟机,使用了ROCm框架,训练速度提升为一局1、2分钟,但其实速度还是很慢。这两天实验室老师拿来了一台主机,配置:Nvidia 显卡,GeForce GTX1660。所以就想安一下GPU,安了一下午,终于可以调用GPU训练了,速度为1分钟12局。。。真是舒服了。1.查看你...原创 2019-09-04 18:01:02 · 3639 阅读 · 2 评论