pytorch环境配置

目录

摘要

Abstract

1.前言

2.安装Anaconda

环境配置

3.安装CUDA

[3.1 查看电脑显卡](#3.1 查看电脑显卡)

配置环境变量

4.配置Pytorch


摘要

本周对pytorch进行环境配置,包括下载安装配置环境变量,判断是否成功配置,以及在pycharm中测试使用。

Abstract

This week involves setting up the PyTorch environment, including downloading, installing, and configuring environment variables, verifying successful configuration, and testing its usage in PyCharm.

1.前言

(1) 什么是Anaconda?

1、 Anaconda是一个开源的数据科学平台,它集成了多个数据科学、机器学习、数据处理和可视化的工具和库。

2、 Anaconda提供了一个高效的包管理器conda,可以便捷地安装和管理PyTorch及其相关的依赖库。Anaconda并不是Pytorch的一部分,但它是管理和安装Pytorch及其依赖项的强大工具。

(2) 什么是Pytorch?

PyTorch是一个开源的深度学习框架,由Facebook的人工智能研究小组开发,主要用于构建和训练神经网络模型。它以其灵活性、易用性和动态计算图的特性在研究和工业界得到了广泛应用。

(3) 什么是PyCharm?

PyCharm是由JetBrains开发的一款集成开发环境(IDE),专门用于Python编程。它功能强大,提供了许多工具和功能,帮助开发者更高效地编写、调试和管理Python项目。

2.安装Anaconda

在官网进行下载,双击下载好的安装文件,开始安装Anaconda。

Download Anaconda Distribution | Anaconda

安装引导页面点击【Next】

点击【Agree】

点击【Next】

选择安装路径

注意!!!:由于Anaconda安装文件较大,强烈建议不要安装在C盘

示例:D:\software\anaconda3

点击 【Next】

高级选项设置:保持默认勾选项,点击【Install】开始安装

等待安装完成(安装过程中建议不要进行其他操作)

安装完成后点击【Next】,然后【Finish】结束安装

环境配置

搜索并打开【编辑系统环境变量】

点击【高级】选项卡,点击【环境变量】

点击【PATH】然后点击【编辑】

注意:用户变量的【PATH】与系统变量的【PATH】配置一个就可以

填写环境变量,这里写你实际安装的目录,图中是示例安装路径。

配置完后点击 【确定】保存

输入cmd打开命令行,打开Anaconda Prompt输入conda --version 或者 conda

3.安装CUDA

如果你有NVIDIA的显卡,就可以进行深度学习。如果没有NVIDIA显卡,则无法使用GPU进行深度学习。

3.1 查看电脑显卡

点击右键【此电脑】→点击右键【管理】→【设备管理器】→【显示适配器】

推荐下载CUDA 11.8版本,不要下载太高的版本。因为CUDA版本需要与PyTorch版本相互匹配,如果CUDA版本太高,可能无法与PyTorch匹配。

CUDA官网 https://developer.nvidia.com/cuda-toolkit-archive

选择下载选项:

  • 操作系统:Windows

  • 架构:x86_64

  • 版本:Windows 10或Windows 11(根据你的系统选择)

双击CUDA安装文件,点击【OK】(使用默认的临时文件夹)

软件许可协议界面,点击【同意并继续】

选择【自定义安装】,点击【下一步】

取消不需要的组件,第四个选项不需要安装,点击下一步

安装位置选择默认位置(C盘),点击下一步

点击下一步开始安装,安装完成后点击关闭

配置环境变量

可以参考上面Anaconda 环境变量的配置

  1. 右键点击"此电脑",选择"属性"
  2. 点击"高级系统设置"
  3. 点击"环境变量"
  4. 在系统变量中找到"Path",双击打开
  5. 添加以下四条路径(假设CUDA安装在C盘):
  6. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA
  7. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin
  8. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\libnvvp
  9. C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\x64

添加完成后,点击确定保存设置。

验证CUDA是否安装成功:

  1. 右键点击"运行",输入CMD并确定
  2. 在命令行中输入:nvcc -V
  3. 如果显示CUDA的版本信息,说明CUDA已经安装成功

4.配置Pytorch

进入PyTorch官网https://pytorch.org/get-started/previous-versions/

找到与CUDA 11.8匹配的PyTorch版本,如PyTorch 2.3.1

复制conda安装命令(不要复制pip安装命令)

接下来在Anaconda Prompt中执行以下步骤:

创建虚拟环境:

conda create -n pytorch2.3.1 python=3.9

输入y确认创建

进入PyTorch虚拟环境:

conda activate pytorch2.3.1

粘贴并执行PyTorch安装命令(使用之前复制的命令),等待下载和安装完成(如果下载中断,可以重新粘贴命令继续下载)

验证PyTorch安装

在PyTorch虚拟环境中,输入Python进入Python解释器,输入以下代码:

python 复制代码
   import torch
   print(torch.cuda.is_available())

如果输出True,说明PyTorch已经安装成功并能够使用GPU

相关推荐
mtouch333几秒前
多源矢量数据接入与API动态聚合:三维可视化沙盘系统的设计与实现
人工智能·无人机·虚拟现实·电子沙盘·数字沙盘
richdata6 分钟前
鞋服品牌应该如何测算AI商品决策项目的ROI?
大数据·人工智能
AI天行健12 分钟前
星宇智算发布无限画布空间索引与语义检索架构技术白皮书
人工智能
hpoenixf15 分钟前
从 Trace 到可信观测:我们的 Agent 可观测体系怎样一步步建立起来
人工智能·agent
鬼手点金16 分钟前
Agent 的学习机制:后训练、上下文学习与外部化学习
java·人工智能·git·学习·自动化·nanogpt
Summer-Bright18 分钟前
英伟达拟收购Reflection、Step5开源license悬空、Codex CLI守护进程争议 —— AI 软件简报 10.10-10.14
人工智能·开源
Angeldream12327 分钟前
第 4 章 医学影像预处理
图像处理·人工智能·python·医学影像·医学影像处理
Web3&Basketball30 分钟前
实测Gemini4Argon:ERQA 87%登顶长程任务榜
数据库·python·大模型·多模态·gemini·长上下文·推理优化
把我格子衫拿来1 小时前
医学影像重采样:spacing 改了,为什么模型位置也跟着错了?
python·dicom·医学影像·simpleitk·重采样
打工仔折腾 AI1 小时前
用 Dashlet 把极空间 NAS 服务入口整理成统一导航首页的实践
java·开发语言·人工智能·后端·python·rpc·ai agent 实战