首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >通过Anaconda安装PyTorch 1.3

通过Anaconda安装PyTorch 1.3
EN

Stack Overflow用户
提问于 2021-12-07 13:25:04
回答 1查看 972关注 0票数 0

我现在的PyTorch版本是1.10。我想把它降级到1.3,但是conda建议只使用CPU版本。我的GPU是一个GTX 1080 TI。

我现在的设置是

代码语言:javascript
运行
复制
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

当我执行:

代码语言:javascript
运行
复制
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的版本。如何找到GPU版本?

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 2021-12-07 14:34:21

好吧,我需要找到合适的棍棒。为了那个

代码语言:javascript
运行
复制
conda search cudatoolkit

然后,我安装了--我可以在Cuda 9.2和10.0之间选择版本。我

看了一下https://pytorch.org/get-started/previous-versions,猜了一下库达的版本。

就我而言:

代码语言:javascript
运行
复制
conda install cudatoolkit==10.0.130

这将自动安装PyTorch==1.3.1

票数 0
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/70260948

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档