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

相关推荐
程序员杰哥4 小时前
Python自动化测试之线上流量回放:录制、打标、压测与平台选择
自动化测试·软件测试·python·测试工具·职场和发展·测试用例·压力测试
吴佳浩4 小时前
LangChain v1 重大更新讲解⚠⚠⚠
python·langchain·agent
模型启动机5 小时前
黄仁勋GTC开场:「AI-XR Scientist」来了!
人工智能·ai·大模型
k***1955 小时前
自动驾驶---E2E架构演进
人工智能·架构·自动驾驶
Techblog of HaoWANG6 小时前
目标检测与跟踪 (4)- 基于YOLOv8的工业仪器仪表智能读数与状态检测算法实
人工智能·视觉检测·智能制造·yolov8·工业检测·指针式仪表·仪器仪表检测
1***Q7846 小时前
深度学习技术
人工智能·深度学习
KKKlucifer6 小时前
2025 国产化数据分类分级工具实测:国产化适配、多模态识别与动态分级能力深度解析
人工智能·分类·数据挖掘
虹科网络安全6 小时前
从AI模型到云生态:构建系统化的企业AI安全管理体系【系列文章(3)】
人工智能·安全
互联网江湖6 小时前
这个Q3,百度开始AI
人工智能·百度