如何在windows wsl上安装pytorch 0.3



如何在windows wsl 2 (Ubuntu-20.04)上安装支持gpu的pytorch 0.3 ?我在这个链接https://pytorch.org/get-started/previous-versions/中尝试了旧版本的pytorch,但无法得到一个可行的解决方案。

我尝试了pip install torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl命令,但它给出了ERROR: torch-0.3.0.post4-cp36-cp36m-linux_x86_64.whl is not a supported wheel on this platform.错误。我使用conda和python 3.6.

在尝试了许多方法之后,事实证明我的GPU支持的最低CUDA版本与PyTorch 0.3不兼容。我有一个RTX 2080s GPU,它具有图灵架构,计算能力为7.5,支持CUDA 10。然而,PyTorch 0.3没有针对CUDA 10编译的预构建二进制文件。(旧版本可以通过官方PyTorch网站获得)因此,如果PyTorch中没有针对目标CUDA版本编译的二进制文件,则必须从PyTorch的源代码构建。当我试图从源代码构建PyTorch 0.3时,它遇到了一个错误,无法构建。然后,在这个阶段,事实证明PyTorch 0.3本身不兼容CUDA 10及以上版本。然而,我成功地在谷歌Colab中运行了用Pytorch 0.3编写的代码,因为Colab给你的GPU是带有CUDA 9的特斯拉K80。有一个针对CUDA 9构建的PyTorch二进制文件。所以在协作中运行代码是可能的。但是如果你想在较新的gpu上运行这样的代码,恐怕这是不可能的。唯一的解决方案是切换到支持新gpu的新版本Pytorch。

关于问题的进一步阅读可以在这里找到。https://discuss.pytorch.org/t/pytorch-0-3-build-from-source-with-cuda-10-cmake-error-cuda-cublas-device-library-advanced/135117

最新更新