Tensorflow或Pytorch是否可以在RTX 20xx super系列上使用

时间:2019-07-18 06:18:59

标签: tensorflow deep-learning cuda pytorch

我将购买新的GPU来学习深度学习。新的Nvidia RTX 2060 Super似乎最符合我的需求。但是我想知道它现在是否与CUDA和tensorflow或pytorch兼容?

3 个答案:

答案 0 :(得分:1)

几天前我购买了相同的VGA并使用它。 经过测试,它在Python(我使用numby库)和CUDA 10.1

中运行良好

答案 1 :(得分:0)

我购买了RTX2070 super,但在设置张量流时遇到了一些麻烦。我遵循以下指南:

https://medium.com/@cjanze/how-to-install-tensorflow-with-gpu-support-on-ubuntu-18-04-lts-with-cuda-10-nvidia-gpu-312a693744b5

*请确保您安装了最新的驱动程序(我的驱动程序是430.40; 430.26没有正确显示我的GPU名称)以及CUDA 10.0和NOT 10.1 ...

**我在验证cuDNN正常工作时遇到了一个错误...请遵循从({https://devtalk.nvidia.com/default/topic/1025801/cudnn/cudnn-test-did-not-pass/)复制的以下步骤:

打开

/usr/include/cudnn.h

并更改行:

#include "driver_types.h"

收件人:

#include <driver_types.h>

***编辑:刚刚安装了pytorch,它似乎也可以正常工作

答案 2 :(得分:-1)

seems like RTX系列支持与CUDA 10兼容的计算能力7.5。使用此卡可以进行深度学习。