CUDA11.6条件下,安装Pytorch的一些问题

在做深度学习项目时,遇到了torch.cuda.is_available()返回false的问题,经CMD查看自己的CUDA版本为11.6。

然后按照该博主的方法操作,https://blog.csdn.net/qq_46126258/article/details/112708781,在该文章的指导下发现自己下载的是CPU版本的Pytorch,于是卸载,打算安装GPU版本。

首先在pytorch官网,选择了pip方式。

然后复制该指令进行下载,但是出现报错 ERROR: Could not find a version that satisfies the requirement torch (from versions: none)……

此时我怀疑是否是因为我添加的镜像网站产生了干扰,于是将镜像源删除,恢复了默认的下载源。

之后可以正常下载了,但是我发现下载的居然还是CPU版本的pytorch (+_+)?

所以我尝试了conda版本的指令发现还是下载的CPU版本的pytorch (#°Д°)

然后我又尝试了一次torch.cuda.is_available()发现结果还是False。

因为上面引用的那个文章的博主的CUDA版本较低,他可以直接从给定的指令中分析所要下载的版本,可是CUDA11.6的版本就看不出来了,此时进入上图中的网址https://download.pytorch.org/whl/cu116

进入torch

经过观察我发现我下载的始终是如下这个版本

观察了一会儿我发现貌似windows系统下,torch的更高版本似乎都没有cp36,也就是python3.6的版本。我所创建的虚拟环境用的是python3.6,此时我怀疑是否是python版本的问题,于是查找在conda环境下升级python版本的方法,无果。

没办法,我只能重新创建了一个虚拟环境,python版本换成了3.9。

再次执行下载好的pip指令,终于好了!!!(o゜▽゜)o☆(创建虚拟环境python版本最好还是高一些)

在下载的过程中,没有遇到下载慢的问题,可能是我挂着VPN的缘故?

至于解决办法的原理我也不太懂,如果有大佬知道可以说一下( •̀ ω •́ )✧

以上就是一个小白在安装pytorch时遇到的一些问题,可能有点基础,也算是记录一下了,说不定可以为之后有相同问题的人提供一些帮助,其他的问题应该可以在上面的文章中找到答案。

感谢@Nefu_lyh

物联沃分享整理
物联沃-IOTWORD物联网 » CUDA11.6条件下,安装Pytorch的一些问题

发表评论