【解决】PyTorch不再支持此GPU,因为它太旧了;源代码编译PyTorch
不二 • 8127 次浏览 • 0 个回复 • 2020年11月27日

用户提问内容:
PyTorch no longer supports this GPU because it is too old.
请问如何能知道pytorch支持哪些显卡?
如何解决:
根据官方的回答,具体可以前往PyTorch不再支持此GPU,因为它太旧了。:
为了减少预编译的二进制文件的大小(已淘汰),我们必须删除少数人使用的一些旧GPU架构。
您仍然可以通过从源代码进行编译来使用它,并且仍然可以像以前一样工作!
就是您可以从源代码进行编译
以获得GPU支持
。据我所知,二进制文件是为cuda计算能力>= 5
而构建的。
编译方法
如果从源代码安装,则将需要Python 3.6.2
或更高版本以及C ++ 14
编译器。另外,我们强烈建议您安装Anaconda
环境。您将获得高质量的BLAS库(MKL),并且无论您的Linux发行版如何,您都可以获得受控的依赖项版本。
一旦安装了Anaconda,请按照以下说明进行操作。
如果要使用CUDA支持进行编译,请安装
- NVIDIA CUDA 9.2或以上
- NVIDIA cuDNN v7或更高版本
- 与CUDA兼容的编译器注意:有关各种支持的CUDA,CUDA驱动程序和NVIDIA硬件的cuDNN版本,您可以参考cuDNN支持列表。
- 如果要禁用CUDA支持,请导出环境变量USE_CUDA=0。可在中找到其他可能有用的环境变量setup.py。
如果您要为NVIDIA的Jetson平台(Jetson Nano,TX1,TX2,AGX Xavier)而构建,请在此处获得安装PyTorch for Jetson Nano的说明。
原文参考:PyTorch no longer supports this GPU because it is too old.
原创文章,转载请注明 :【解决】PyTorch不再支持此GPU,因为它太旧了;源代码编译PyTorch - pytorch中文网
原文出处: https://ptorch.com/news/257.html
问题交流群 :168117787
- 没有评论