我买了一块GTX 1660TI显卡，显卡驱动10.1，请问能使用caffe、tensorflow训练模型吗？

Question

我买了一块GTX 1660TI显卡，显卡驱动10.1，请问能使用caffe、tensorflow训练模型吗？官网并没有给出计算能力，试了N次，一直显示不支持，请问需要怎么修改。

Answer 1

老黄去年下禁令禁止游戏显卡跑深度学习。解决方法如下：（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,...)

Answer 2

理论上是可以的。
虽然没试过GTX1660TI，但是GTX1060是没有问题的，新款显卡没有理由不支持，如果楼主测试，加速环境显示不支持，可以更换别的编译器版本，也可以参考网上的深度学习环境搭建参考一下。

Answer 3

请问您弄好了吗。我也想知道

Answer 4

请问您弄更好了吗

Answer 5

不懂你说的什么训练模式