通过 Anaconda 安装 PyTorch 1.3

Question

我目前的 PyTorch 版本是 1.10。我想将它降级到 1.3，但 conda 建议使用 CPU-only 版本。我的 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版本？

Answer 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。

通过 Anaconda 安装 PyTorch 1.3

Install PyTorch 1.3 via Anaconda

pytorch

anaconda