本文为自己整理的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安装成功。