我试图在我的GPU上运行代码,首先我遇到了在ubuntu 18.04上运行CUDA的问题。在经历了一些痛苦之后,有人建议我在Docker Image中工作,在那里(我假设)我有一个工作良好的CUDA版本(我正在开始学习pytorch,并且我能够获得一个torch.cuda.is_available我收到消息: NotImplementedError: bounds checking is not supported for CUDA 我真的不能理解一个合适的方法来修复它。-----------------
经过一番搜索,我发现这个问题可能是由旧的CUDA版本引起的。(found suitable version "7.5", minimum required is "6.5") -- Found gcc >=5 and CUDACompiling without MAGMA support-- got cuda version 7.5
--Found <
I pip安装了tensorflow-GPU2.4.1我还安装了CUDA11.2和cudnn11.2,遵循中的步骤,还检查了所有路径是否正常以及库是否在正确的位置。我还运行了tf.test.is_built_with_cuda(),它返回True。
那么问题是我的GPU不支持当前版本的CUDA或cudnn吗?