☰
2024年 pytorch安装教程(GPU版)
2026/10/28 18:31:49 网站建设 项目流程

一、首先安装好 Anaconda

(此处不详细介绍,如果有没有安装Anaconda的,可以看一下下面我发的清华镜像网站,自行下载,注意:不建议下载最新版本)

Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

二、安装 pytorch

1.先打开 anaconda prompt

2.创建虚拟环境

conda create -n 虚拟环境名字 python=你要下载的版本

以我的为例:

2.1创建虚拟环境,我的是 python3.9

conda create -n pytorch390 python=3.9

2.2 输入 y

2.3 安装完成后,输入 conda env list 检验是否安装成功。

conda env list

3.进入刚才我们已经创建好的虚拟环境

输入:

conda activate pytorch390

4.查看CUDA版本

输入:

nvidia-smi

因为我的CUDA是12.4版本,所以等下我只需要下载一个比12.4版本低的就可以了。

5.搜索 pytorch官网 或者 点解我下面我的链接也可以

PyTorchhttps://pytorch.org/进入pytorch官网,因为我的CUDA是12.4版本,所以我下载的版本是CUDA12.1

复制下面的这个代码

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121

回到 Anaconda prompt,将上面的这串代码复制进去:

输入:y

等待它安装成功即可。

至此,pytorch安装结束。

三、验证pytorch安装是否成功

首先,输入 python ,进入 python 环境

python

然后,输入 import torch

import torch

最后,输入 torch.cuda.is_available(),返回 ture 说明安装成功。

torch.cuda.is_available()

需要专业的网站建设服务?

联系我们获取免费的网站建设咨询和方案报价,让我们帮助您实现业务目标

立即咨询