我现在的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的版本。如何找到GPU版本?
发布于 2021-12-07 14:34:21
好吧,我需要找到合适的棍棒。为了那个
conda search cudatoolkit
然后,我安装了--我可以在Cuda 9.2和10.0之间选择版本。我
看了一下https://pytorch.org/get-started/previous-versions,猜了一下库达的版本。
就我而言:
conda install cudatoolkit==10.0.130
这将自动安装PyTorch==1.3.1
。
https://stackoverflow.com/questions/70260948
复制相似问题