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

2024年11月15日 02:59
有5个网友回答
网友(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,...)

网友(2):

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

网友(3):

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

网友(4):

请问您弄更好了吗

网友(5):

不懂你说的什么训练模式