当前位置:   article > 正文

pytorch搭建Python3.7_python3.7对应的pytorch版本

python3.7对应的pytorch版本

Python版本:3.7

1.安装任意版本的Anaconda

首先安装anaconda Windows下安装anaconda

进入文件夹,选择对应的anaconda版本,选择如下版本为python3.7(要安装其他的就选其他版本号)
Anaconda其它版本文件夹

s
安装完成后,在搜索框里输入cmd,打开terminal看看有木有安装成功

输入 python -V
s
当当当当当当~ 3.7版本安装成功啦
s

2.配置显卡NVIDIA

打开任务管理器,ctr+Alt+Del,然后看看是否显示是它,是就说明驱动已经存在的,不是的话去安装一下驱动就行。
如果没有GPU的话,注意哈,没有的话直接省略此步鄹,直接进入步鄹4
s

3.查看cuda版本

打开NVIDIA控制面板,咋个打开呢,可以在搜索框里输入NVIDIA

打开后选择系统信息,点击组件,查看对应的版本信息。
s
不需要自己去手动安装CUDA和cuDNN,pytorch官方都弄得非常到位了。只需要去官网copy一下命令就行了。

4.创建一个pytorch存放仓库

创建它的原因是将不同版本的pytorch隔开,方便使用。打开anaconda prompt,创建一个叫做mytorch的仓库,也可以叫其它名字
conda create -n mytorch python=3.7

在这里插入图片描述
然后输入y 回车

安装完成后输入

conda info --envs
再输入 conda activate mytorch
激活当前环境并进入刚刚创建的仓库
s

5.安装pytorch

进入pytorch官网
下拉看到该页面,按照自己计算机的版本选择对应的pytorch版本号
我的电脑支持的CUDA版本是11.1.114
进入之前版本号选择
注意,前面没有GPU的这里直接选择CPU即可,不需要再去安装什么CUDA了

s
在刚才的命令行里输入对应的命令

conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
  • 1

s
输入y回车,耐心等待,完成后是这个样子
s

6.验证

接下来是验证时刻,如下图所示则安装成功

#先输入
python
#再输入
import torch
#然后输入
import torchvision
#最后输入
torch.cuda.is_available()
  • 1
  • 2
  • 3
  • 4
  • 5
  • 6
  • 7
  • 8

s

声明:本文内容由网友自发贡献,转载请注明出处:【wpsshop】
推荐阅读
相关标签
  

闽ICP备14008679号