mxnet吧 关注:41贴子:32
  • 6回复贴,共1

mxnet的GPU版本有cuda=11.0的版本吗?我尝试下

只看楼主收藏回复

mxnet的GPU版本有cuda=11.0的版本吗?我尝试下载conda install mxnet-cu110,提示没有找到,然后下载了9.0版本的,无法使用。有没有大佬知道mxnet有没有cuda=11.0的版本呢?


IP属地:江苏来自Android客户端1楼2020-09-25 09:08回复
    你如果用的是linux系统(比如Centos7 ubuntu等)那么可以在这个网站https://dist.mxnet.io/python找到官方预训练的支持cuda11.0的各版本的mxnet包


    IP属地:四川2楼2021-01-08 16:13
    收起回复
      谢谢,但是楼主使用的是Windows。这个问题最终的解决是弃mxnet而取pytorch了


      IP属地:江苏来自Android客户端3楼2021-08-07 20:21
      回复