pytorch安装时默认安装CPU版本

在pytorch官网,选好对应的CUDA之后,根据官网下提供的命令(红框)运行,开开心心的下载好了相应版本的pytorch,激动的使用torch.cuda.is_available()去验证,心想终于配置好了对应GPU版本的pytorch,谁知却返回了false
pytorch安装时默认安装CPU版本_第1张图片
这时候你可以去看一下,问题就在这,安装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

你会发现,下图框出来的部分是py3.7_cpu…,类似于这样子
pytorch安装时默认安装CPU版本_第2张图片

解决方法如下:

conda uninstall cpuonly

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

你可能感兴趣的:(问题记录,pytorch,深度学习,python)