深度学习之pytorch安装(二)

安装pytorch模块

这里是安装pytorch的第二个步骤,正式进入了pytorch模块。
首先,如果你想要使用pytorch,那么你的电脑上面必须要有英伟达的显卡,那么怎么看自己电脑上面有没有英伟达的显卡呢?这里可以去看我的另一篇文章,查看自己电脑上面是否由英伟达的显卡。

https://blog.csdn.net/congcong0509/article/details/127753424?spm=1001.2014.3001.5501

第二步,安装pytorch模块

但是在安装pytorch模块之前,先需要确定一下自己电脑的配置。
查看自己电脑CUDA的版本,windows+R,输入cmd命令,打开终端命令窗口。

输入:nvidia-smi
在这里插入图片描述
这里可以看到,我的CUDA版本是12.0(这个版本我估计是最高了的),CUDA版本不是越高越好,因为CUDA越高,所需要安装的CUDA驱动的版本也好很高。
注意:CUDA安装的驱动版本要高于CUDA的版本。

通过官网命令进行pytorch的安装。下面介绍的就是通过官网命令进行安装的。

好了,下面进行pytorch的安装流程。
1、在安装完成Annconda之后,进入到Annconda Prompt应用里,点击打开。
在这里插入图片描述
2、进入之后,创建一个自己的虚拟环境,这个虚拟环境里主要存放pytorch的相关的一些包。

输入:conda create -n 虚拟环境名称 python=3.8 -c 镜像源
这里提供一些安装的镜像源,以加速安装。

清华源:https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main

阿里云:http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/

在这里插入图片描述
3、输入y,回车。
在这里插入图片描述
4、在安装的过程中,我们可以打开浏览器,在地址栏中输入

pytorch.org
然后回车,会进入到pytorch的官方网站。
在这里插入图片描述
5、点击,安装。
在这里插入图片描述
6、点击安装之前的版本。
在这里插入图片描述

7、选择CUDA11.3的版本进行安装。
好了,到此回到安装虚拟环境的终端窗口。
在这里插入图片描述
好了,上面就表示虚拟环境安装好了,并且我们指定了一个python编译器python3.8。

8、进入到自己创建的虚拟环境中。

输入:conda activate 虚拟环境名称

回车之后,如下所示。
在这里插入图片描述
在这里插入图片描述
9、安装pytorch

输入:pip install torch1.12.1+cu113 torchvision0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

在这里插入图片描述
上面的安装过程比较慢,大约1小时左右。大家可以先去学习或者去休息一会,一个小时之后再回来进行查看。

10、安装好之后,进行查看。
在这里插入图片描述
正如上图所示,输入一下命令。

python
import torch
torch.cuda.is_available()

如果依次输入上面的命令回车之后,显示True的话,那么恭喜你,pytorch就安装好了。
好了,这就是第一种的安装pytorch的方法。这种方法虽然缓慢,但是对于出血pytorch的人来说,等待也许就是一种好的方法。

那么下面就来看一下第二方方式:百度网盘链接下载。我按照的pytorch版本是10.2的,这里给个建议
大家按照pytorch的话建议按照10.0+和11.3版本的torch包。这是因为这两个版本的torch包可以兼容大部分的CUDA版本。如果按照最新的torch包(像是11.7、11.8)也许你会遇到各种各样的问题。所以建议按照这两个兼容性较好的版本!

下面提供几条简单的conda命令:
创建虚拟环境:conda create -n 虚拟环境名称 python=指定python解释器的版本 -c 镜像源
删除虚拟环境:conda remove -n 虚拟环境名称 --all
进入到虚拟环境:conda activate 虚拟环境名称
退出当前虚拟环境:conda deactivate

猜你喜欢

转载自blog.csdn.net/congcong0509/article/details/127877358
今日推荐