通过 Anaconda 安装 PyTorch 1.3答案

【问题标题】：Install PyTorch 1.3 via Anaconda通过 Anaconda 安装 PyTorch 1.3
【发布时间】：2022-01-12 13:47:23
【问题描述】：

我当前的 PyTorch 版本是 1.10。我想将其降级到 1.3，但 conda 建议仅使用 CPU 版本。我的 GPU 是 GTX 1080 TI。

我目前的设置是

cudatoolkit               10.2.89         hfd86e86_1  
cudnn                     7.6.5           cuda10.2_0  
pytorch                   1.10.0          py3.7_cuda10.2_cudnn7.6.5_0    pytorch
pytorch-mutex             1.0             cuda    pytorch

当我执行时：

conda install pytorch==1.3.0 torchvision -c pytorchCollecting package metadata (current_repodata.json): done
Solving environment: done

## Package Plan ##

  environment location: /home/lorenzp/.conda/envs/detection

  added / updated specs:
    - pytorch==1.3.0
    - torchvision


The following packages will be downloaded:

    package                    |            build
    ---------------------------|-----------------
    cpuonly-1.0                |                0           2 KB  pytorch
    pytorch-1.3.0              |      py3.7_cpu_0        36.1 MB  pytorch
    torchvision-0.4.1          |         py37_cpu        14.6 MB  pytorch
    ------------------------------------------------------------
                                           Total:        50.7 MB

The following NEW packages will be INSTALLED:

  cpuonly            pytorch/noarch::cpuonly-1.0-0

The following packages will be DOWNGRADED:

  pytorch                1.10.0-py3.7_cuda10.2_cudnn7.6.5_0 --> 1.3.0-py3.7_cpu_0
  torchvision                             0.11.1-py37_cu102 --> 0.4.1-py37_cpu


Proceed ([y]/n)? n


CondaSystemExit: Exiting.

它只是建议 CPU-only 版本。如何找到 GPU 版本？

【问题讨论】：

标签： pytorch anaconda

【解决方案1】：

嗯，我需要找到合适的 cudatoolkit。为此

conda search cudatoolkit

然后，我安装了我可以选择 Cuda 9.2 和 10.0 之间的某个版本。我
看了https://pytorch.org/get-started/previous-versions，猜了一下cuda版本。

就我而言：

conda install cudatoolkit==10.0.130

这会自动安装PyTorch==1.3.1。

【讨论】：