老黄去年下禁令禁止游戏显卡跑深度学习。解决方法如下:(tensorflow)
方法一:改变系统变量使得仅目标显卡可见:
1 export CUDA_VISIBLE_DEVICES=0 #这里是要使用的GPU编号,正常的话是从0开始
方法二:使用tensorflow api添加设备环境
1 with tensorflow.device('/gpu:0'):
2 main tensorflow program
注意:
如果使用方法二,需要修改一下Session配置,否则在默认配置下,所有可见显卡的显存都会被占用(暂时还没找到原因)。
修改代码:
1 config=tensorflow.ConfigProto()
2 config.gpu_options.allow_growth=True
3 ...
4 sess=tensorflow.Session(config=config,...)
理论上是可以的。
虽然没试过GTX1660TI,但是GTX1060是没有问题的,新款显卡没有理由不支持,如果楼主测试,加速环境显示不支持,可以更换别的编译器版本,也可以参考网上的深度学习环境搭建参考一下。
请问您弄好了吗。我也想知道
请问您弄更好了吗
不懂你说的什么训练模式