【问题标题】:Can we install Pytorch CUDA 11.3 when the system has CUDA 11.2当系统有 CUDA 11.2 时我们可以安装 Pytorch CUDA 11.3
【发布时间】:2022-11-05 07:37:40
【问题描述】:

我的 PC 中有 cuda 11.2,想安装 PyTorch。 PyTorch 在 website 中只提到了 CUDA10.2 和 11.3

我可以在我的电脑上安装 torch==1.10.1+cu113 吗? 如果没有,我该如何为 CUDA11.2 安装 PyTorch

我不想更改我的 CUDA 版本,因为我有其他应用程序正在使用它。

【问题讨论】:

  • 为什么不试试看会发生什么?
  • 您必须安装 Pytorch 所针对的 CUDA 版本。这是一个不可协商的要求。您的两个选择是为预构建的二进制文件安装第二个 CUDA 运行时 (CUDA 11.3),或者使用 CUDA 11.2 构建您自己的 Pytorch

标签: pytorch


【解决方案1】:
pip install torch==1.7.1+cu110 torchvision==0.8.2+cu110 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html

这有效,但不是最新版本

【讨论】:

    【解决方案2】:

    我尝试了 11.3 的那个,到目前为止它在 GPU 上运行良好:

    sudo pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
    

    【讨论】:

      【解决方案3】:

      当然可以。

      先卸载当前的cuda
      须藤/usr/local/cuda-X.Y/bin/cuda-uninstaller

      并使用 dpkg 卸载 libcudnn
      例如:sudo dpkg --remove libcudnn7

      然后下载runfile并安装它还安装相应的libcudnn版本

      【讨论】:

        猜你喜欢
        • 1970-01-01
        • 2021-04-27
        • 2022-10-31
        • 2019-02-17
        • 2021-02-27
        • 2023-02-26
        • 1970-01-01
        • 2020-06-07
        • 2021-06-18
        相关资源
        最近更新 更多