torch官网安装GPU版本输出为CPU

一、按照官网的命令行安装torch=1.10.0：

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

时出错：

conda list检查也发现没有问题：

二、解决方案：

最后通过: nvidia-smi

没有正确的显示,

正确应该是这样的

或者通过检查附加驱动,发现勾选的是最后一个

正确版本应如下:

(不知道抽什么风没用自带的软件更改,用命令行改,结果电脑又黑屏开不了机,又只能再度重装!!!!)

检查发现可能是本机的显卡驱动没弄好

三. 总言之, 把本机上的显卡驱动弄好之后

按照官网给的语句选择适合的版本就好了

官网: Previous PyTorch Versions | PyTorch

安装语句:

conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge

四\ 检查输出

import torch

print(torch.__version__)

print(torch.cuda.device_count())

print(torch.version.cuda)

print(torch.cuda.is_available())

又出现这个问题了，这次可能不是驱动安装的问题，这次驱动检查没问题。