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

Posted zhendelanQAQ

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了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

Windows 下安装 CUDA 和 Pytorch 跑深度学习

Windows 下安装 CUDA 和 Pytorch 跑深度学习

一、安装cuda 11.3

dxdiag

有独立显卡,支持安装。

1.1 download cuda 11.3

1.2 双击解压


参考:《动手学深度学习v2》

Installed:
     - Nsight Monitor
Not Installed:
     - Nsight for Visual Studio 2019
       Reason: VS2019 was not found
     - Nsight for Visual Studio 2017
       Reason: VS2017 was not found
     - Integrated Graphics Frame Debugger and Profiler
       Reason: see https://developer.nvidia.com/nsight-vstools
     - Integrated CUDA Profilers
       Reason: see https://developer.nvidia.com/nsight-vstools

1.3 测完安装是否成功

nvidia-smi

二、安装Anaconda

2.1 下载安装

官网:https://www.anaconda.com/

2.2 查看python版本

python --version

三、安装Pytorch

去pytorch.org下载它的GPU版本:https://pytorch.org/

3.1 安装

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113


安装完成后测试一下,cuda 安装是否成功

3.2 安装jupyter d2l包

pip isntall jupyter d2l

3.3 启动jupyter

jupyter notebook

在使用前,请关闭本机的独显,使用核显!!!!!
在使用前,请关闭本机的独显,使用核显!!!!!
在使用前,请关闭本机的独显,使用核显!!!!!

不然会报内存不足等错误!

运行环境

电脑:拯救者Y9000P
显卡: RTX3060

以上是关于CUDA11.6条件下,安装Pytorch的一些问题的主要内容,如果未能解决你的问题,请参考以下文章

RTX3090+win10+CUDA11.6+cudnn8.5.0+pytorch1.12.1 环境——个人配置经验

深度学习常用PyTorch CUDA版本whl下载及在线安装命令

pytorch安装报错:OSError: [WinError 127],c10_cuda.dll缺失。

第三篇 Ubuntu 20.04 搭建AI开发环境

安装tensorflow的GPU版本(详细图文教程)--CUDA11.6的安装

安装tensorflow的GPU版本(详细图文教程)--CUDA11.6的安装