PyTorch环境配置及安装

PyTorch环境配置及安装

Step1:安装Anaconda

参考该链接(视频01:30--03:00为安装教程):

【PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】】 https://www.bilibili.com/video/BV1hE411t7RN/?share_source=copy_web&vd_source=c71860a35ae6f457f141a44ee4b38133

Step2:创建环境

安装Anaconda之后,可通过Anaconda Prompt进入创建好的环境中,默认为base环境,可以输入pip list查看环境中已安装的包

每个工具包都有其对应的版本,例如一个项目要用到PyTorch0.4,另一个项目要用到PyTorch1.0,如果只使用base环境,则需要反复卸载、安装工具包以满足项目需求,费时费力。

使用Anaconda集成的conda包(安装Anaconda后,可直接使用)能够解决这个问题,它可以新建环境,一个环境安装PyTorch0.4,一个环境安装PyTorch1.0,可根据需要切换环境。

  • 创建环境的指令为:conda create -n pytorch python=3.10
    • conda是指调用conda包,create是指创建,-n pytorch是指环境名为pytorchpython=3.6是指创建的环境中python版本为3.10
  • 切换环境的指令为:conda activate pytorch


Step3:查看GPU型号

  • 方法1:任务管理器------性能------GPU(如果有的话,可以看到型号)
  • 方法2:设备管理器------显示适配器

Step4:安装PyTorch

PyTorch官网:https://pytorch.org

  • 根据需求确定安装命令
    • 一般建议安装最新的稳定版本(Stable),也可以点击Previous versions of PyTorch选择要安装的版本;
    • 若有GPU,则应安装CUDA版本,否则应安装CPU版本
    • 示例:conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
  • 复制命令到已创建的环境中(环境名为pytorch),执行即可安装PyTorch
  • 查看安装结果
    • 输入pip list,结果中有torch基本说明安装成功;
    • 输入python,进一步输入import torch,不报错说明安装成功;
    • 进一步检查是否可以使用GPU,输入torch.cuda.is_available(),若返回True说明torch能正常使用GPU

安装时的常见问题(安装比较慢、安装后不能使用GPU)及解决方案见该链接:

【PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】】 https://www.bilibili.com/video/BV1hE411t7RN/?p=3&share_source=copy_web&vd_source=c71860a35ae6f457f141a44ee4b38133

  • 若使用命令安装比较慢
    • 将安装包下载到本地(C:\Users\20458\anaconda3\pkgs文件夹下),使用对应的命令进行安装(打开Anaconda Prompt输入命令conda install --use-local 要安装的包名
  • 若不能正常使用GPU
    • 检查电脑显卡型号,确认是否支持GPU
    • 检查显卡驱动(打开Anaconda Prompt输入命令nvidia-smi),其版本号不能低于torch版本的要求,若低于要求则应更新显卡驱动

Step5:安装PyCharm

说明:社区版(Community Condition)基本能够满足日常使用的需求

参考该链接(视频01:00--03:00为安装教程):

【PyTorch深度学习快速入门教程(绝对通俗易懂!)【小土堆】】 https://www.bilibili.com/video/BV1hE411t7RN/?p=2&share_source=copy_web&vd_source=c71860a35ae6f457f141a44ee4b38133

Step6:使用PyCharm创建项目

  • Location为项目地址,图中项目名为pytorchPractice
  • 选择已经有的编译器(Previously configured interpreter):选择新创建的环境中的编译器(可在Anaconda的安装目录下选择编译器C:\用户\XXX\anaconda3\envs
  • 检查torch是否能正常引入,以及是否能调用GPU查看安装结果(选择以下任一即可)
    • 使用脚本文件(右键项目pytorchPractice------New------Python File,创建脚本文件):执行对应代码,查看结果;
    • 使用Python Console(控制台:交互式即时执行代码):执行对应代码,查看结果
相关推荐
ECT-OS-JiuHuaShan2 分钟前
否定之否定的辩证法,谁会不承认?但又有多少人说的透?
开发语言·人工智能·数学建模·生活·学习方法·量子计算·拓扑学
软件开发技术深度爱好者10 分钟前
基于多个大模型自己建造一个AI智能助手(增强版)
人工智能
c***871911 分钟前
Flask:后端框架使用
后端·python·flask
骥龙17 分钟前
4.12、隐私保护机器学习:联邦学习在安全数据协作中的应用
人工智能·安全·网络安全
天硕国产存储技术站26 分钟前
DualPLP 双重掉电保护赋能 天硕工业级SSD筑牢关键领域安全存储方案
大数据·人工智能·安全·固态硬盘
腾讯云开发者32 分钟前
AI独孤九剑:AI没有场景,无法落地?不存在的。
人工智能
光影少年35 分钟前
node.js和nest.js做智能体开发需要会哪些东西
开发语言·javascript·人工智能·node.js
落798.37 分钟前
基于CANN与MindSpore的AI算力体验:从异构计算到应用落地的实战探索
人工智能·cann
audyxiao0011 小时前
期刊研究热点扫描|一文了解计算机视觉顶刊TIP的研究热点
人工智能·计算机视觉·transformer·图像分割·多模态
paopao_wu1 小时前
目标检测YOLO[04]:跑通最简单的YOLO模型训练
人工智能·yolo·目标检测