PyTorch环境的配置及安装

1、安装Anaconda
https://repo.anaconda.com/
安装相应的版本,我这里安装的是Anaconda3-5.2.0-Windows-x86_64.exe

在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

2、显卡准备
深度学习离不开显卡(PyTorch支持英伟达显卡)
检查显卡驱动是否正常安装:正常显示表示,驱动正常安装。
在这里插入图片描述

3、管理环境

3.1、
在这里插入图片描述
在这里插入图片描述

3.2、初始环境(base) (初始的屋子)
下面我们创建另一个环境(pytorch):

conda create -n pytorch python=3.6

在这里插入图片描述

3.3、激活环境(pytorch):

conda activate pytorch

在这里插入图片描述

3.4、查看这个环境(屋子)中有哪些包

pip list

在这里插入图片描述
很明显,上面的包中没有pytorch包,所以我们要安装pytorch

3.5.安装pytorch
3.5.1、进入pytorch官网:https://pytorch.org/
英伟达显卡,如下操作:
在这里插入图片描述

选好后,复制运行指令:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

3.5.2、查看驱动版本是否符合conda要求
命令窗口 输入nvidia-smi,查看驱动版本。
conda仅支持驱动版本号大于396.26
在这里插入图片描述

3.5.3,、
在环境(pytorch)中运行3.5.1复制的指令

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

在这里插入图片描述
再查看一下安装的包有哪些

pip list

在这里插入图片描述

很明显,pytorch包成功安装

3.6、检查pytorch是否安装成功
在pytorch环境中 输入 python
输入import torch若没有报错,正常显示,则pytorch成功安装。
在这里插入图片描述

3.7、检查pytorch是否可用使用你电脑的GPU
输入torch.cuda.is_available()
如果显示True,则你的这台计算机的GPU可以给pytorch使用
在这里插入图片描述

至此,大功告成。

4.总结

  • 创建conda环境,指定python版本
  • 激活环境使用
  • 安装pytorch
    在这里插入图片描述
  • 16
    点赞
  • 69
    收藏
    觉得还不错? 一键收藏
  • 4
    评论
评论 4
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值