pytorch安装时默认安装CPU版本_解决pytorch安装时默认安装cpu版本的问题

在pytorch官网，选好对应的CUDA之后，根据官网下提供的命令（红框）运行，开开心心的下载好了相应版本的pytorch，激动的使用torch.cuda.is_available()去验证，心想终于配置好了对应GPU版本的pytorch，谁知却返回了false

这时候你可以去看一下，问题就在这，安装pytorch时会默认安装这个cpuonly库，就是这个函数库导致我们安装的pytorch是CPU版本的，可以使用conda list查看一下自己有没有，如果有的话就是它在作怪
可以回头验证一下，比如重新输入官网提供的下载GPU版本的pytorch命令，比如：

conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=10.2 -c pytorch

1
2

你会发现，下图框出来的部分是py3.7_cpu…，类似于这样子

解决方法如下：

conda uninstall cpuonly
1

卸载之后重新输入命令，再次上看红框的位置，有cuda_cudnn则正确下载了