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

相关推荐
一个java开发3 分钟前
Dask 配置文件加载机制说明
大数据·python
海边夕阳20068 分钟前
【每天一个AI小知识】:什么是大语言模型(LLM)?
人工智能·深度学习·神经网络·机器学习·语言模型·自然语言处理·llm
算力魔方AIPC8 分钟前
破解“竖排文本”魔咒:在 RTX 3060 上微调 PaddleOCR-VL 以识别日本漫画
人工智能
bj_zhb10 分钟前
图片的base64表示
python·llm
飞Link12 分钟前
【Django】Django 调用外部 Python 程序的完整指南
后端·python·django·sqlite
周杰伦_Jay19 分钟前
【Java集合与线程池深度解析】底层原理+实战选型+避坑指南(附代码)
java·开发语言·python
袖手蹲19 分钟前
Arduino UNO Q 从 Arduino Cloud 远程控制闪烁 LED
人工智能·单片机·嵌入式硬件·电脑
doris61023 分钟前
设备点检、保养、维修一站式解决方案
大数据·数据库·人工智能
北京耐用通信24 分钟前
终结混合网络调试噩梦:耐达讯自动化实现EtherCAT对DeviceNet设备的直接读写
网络·人工智能·物联网·网络协议·自动化·信息与通信
BFT白芙堂24 分钟前
Franka机械臂“举一反三”:LLM Trainer如何通过单次演示实现自动化数据生成与长程任务学习
人工智能·学习·机器学习·自动化·模型训练·具身智能·franka