Pytorch 环境配置

wuan2025-03-302025-04-07

前言：配环境还真是技术员们的一大难题，每次配环境看着视频以为很简单，结果自己出现各种各样的错误，我本来是看了小土堆的视频来配置Pytorch视频的,但其实对现在已经不太适用了（仅仅代表个人看法）

Pytorch环境配置

1.下载anaconda3

这个正常在网站上安装就好了，但是要注意盘别乱安装。我就是第一次安装在D，然后后面环境又在C导致错误了。不妨就按照系统的放在C盘，之后也很OK。

如果想使用anaconda的历史版本（https://repo.anaconda.com/)

下载过程可能中间会卡住一段很正常，耐心等待就好了。

2.安装完成后在菜单的Anaconda Prompt 中运行指令：
conda create -n pytorch python=3.9

这个地方为什么是3.9 我觉得有点讲究的我刚开始就是按照视频的3.6，后面发现没用。

后面遇到y/n 一直按y就好了

3.安装好后可能每次进入都是base环境这个时候就需要记住一个指令：

conda activate pytorch

这个时候左边的（base) 就变成了（pytorch)了

环境就变成pytorch环境

取消环境的话

conda deactivate

4.接着就是开始pytorch的安装

这个是官网链接

进去后你就会发现有个

NOTE: Latest PyTorch requires Python 3.9 or later.

所以这就是为什么上面我说python=3.9的原因

小土堆视频是conda安装但是现在好像不能conda安装了，以下是官方文件：

PyTorch官方在2024年底发布公告，宣布不再维护Conda包，原因是Conda的维护成本过高1 2。因此，PyTorch官网安装页面上出现了“Conda packages are no longer available. Please use pip instead.”的提示1。

所以下面我来介绍pip的安装方式：

你可以通过Win+R输入cmd,进入后输入：

nvidia-smi

我的是 CUDA Version: 12.6

复制command即可，如果你想安装其他版本的可以在Previous PyTorch Versions | PyTorch

我下载的时候是有点慢的可以稍微等一等

或者小土堆视频下方有下载链接你可以按照他的视频去操作但感觉不一定适用因为他之前是package是conda.

5.在Pytorch下的时候是：

（pytorch)C:\Users\用户名>

输入

（pytorch)C:\Users\用户名>python

进入python环境后，再输入

import torch

没报错的话稍微等一会会出现>>>

下面来看下pytorch是否能使用GPU

torch.cuda.is_available()

返回True

这样就表示你的电脑的GPU是可以被pytorch使用的。