1.PyTorch环境的配置及安装(Configuration and Installation of PyTorch)

本教程详细介绍了如何安装anaconda并创建PyTorch专用的虚拟环境,包括设置Python版本、检查显卡驱动、管理环境、安装PyTorch及验证安装成功的步骤。通过清华源加速下载,并提供检查GPU可用性的方法。
摘要由CSDN通过智能技术生成

参考PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】

安装anaconda

官网
在这里插入图片描述
视频中所用python版本为3.6,因此需要到anaconda历史版本中找到适配相应版本的安装包
在这里插入图片描述
对应地址

在这里插入图片描述
文件地址可以自己设置,其余选项默认即可
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述

显卡准备

查看显卡驱动是否正确安装了
打开任务管理器
在这里插入图片描述
如果能显示显卡型号则说明显卡驱动已经正确安装了
如果没有则需要去官网下载
在这里插入图片描述

管理环境

conda create -n pytorch python=3.6创建专门的虚拟环境
在这里插入图片描述
conda activate pytorch进入pytorch环境
pip list查看安装的库

安装pytorch

nvidia-smi查看驱动版本
在这里插入图片描述
检查驱动版本Driver Version是否大于396.2
如果版本数小于要求,则可以进入英伟达驱动官网,选择适合自己电脑的驱动程序然后下载更新
在这里插入图片描述

在下载pytorch前先进行换源
anaconda 的清华镜像帮助
在这里插入图片描述

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64

conda config --show channels检查已经加入的channels
在这里插入图片描述

进入pytorch官网
在这里插入图片描述
-c pytorch为强制使用pytorch官网安装,因此只需要执行conda install pytorch torchvision torchaudio cudatoolkit=11.3

如果还是很慢可以考虑用pip加清华源进行安装pip install torch -i https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

测试

输入python进入python环境
输入import torch导入torch库,同时下一行也出现>>>说明pytorch安装成功
在这里插入图片描述
输入torch.cuda.is_available()返回为True说明GPU可以使用
在这里插入图片描述

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值