Pytorch的下载安装

本文为自己整理的Pytorch下载相关的内容笔记，以便日后查阅

一. 基本命令

1.查看conda版本

conda --version

2.创建conda新环境

conda create –n 名称 python=版本

3.查看已经创建的conda环境

conda info --envs

4.进入虚拟环境

conda activate 环境名称

为了避免以前的环境python版本与现在不二. 下载Pytorch适配，我选择了重新建了一个3.10.10版本的环境

conda create -n study python=3.10.10

激活

conda activate study

先看看自己电脑显卡配置

nvidia-smi

我的为 11.7

前往Pytorch官网  PyTorch

我查看了Pytorch以前的版本，找到了与我相适配的版本

# CUDA 11.7
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

二. 安装

在配置的环境中输入上述命令执行即可

conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

三. 验证

使用管理员权限打开anaconda指令框，在指定的conda环境中，输入以下命令

python

然后输入

import torch

会有较长一段时间反应，然后输入

torch.cuda.is_available()

会打印True。

效果如下：

此时则Pytorch安装成功。