torch.cuda()函数问题

时间:2018-05-28 08:55:50

标签: python cuda torch

我是火炬手的新人。当我用TensorFlow运行我的程序时,它正在工作。但这是火炬的问题。 我搜索了很多,但找不到任何有用的答案。谁能帮帮我?

错误: enter image description here 我的环境:

  • windows10 64bit
  • python3.6
  • cuda9.0
  • cudnn64
  • GPU:GTX965m

1 个答案:

答案 0 :(得分:0)

正如您在发行说明here中所看到的那样,Torch开发人员决定从Torch 0.3.1开始,从他们的版本中弃用计算能力3.0和5.0设备。

您的设备是计算机5.0设备,因此在最新版本的Torch中不受支持。

您可以了解哪些替代方案对您开放here(基本上使用较旧版本的Torch,或者从源代码构建您自己的替代品并支持您的GPU)。