Anaconda安装Pytorch2.0.0(GPU版本)步骤

1. 安装版本

pytorch版本:2.0.0

torchvision版本:0.15.0

torchaudio版本:2.0.0

cuda版本:cuda11.7

python版本:3.9

2. 搭建虚拟环境

第一步:创建虚拟环境

打开Anaconda Powershell Prompt ,输入

conda create --name pytorch2.0 python=3.9

 第二步:切换进入虚拟环境

conda activate pytorch2.0

3. 下载pytorch、torchvision以及torchaudio的whl文件

网站:download.pytorch.org/whl/torch_stable.html

Anaconda安装Pytorch2.0.0(GPU版本)步骤_第1张图片

4. 通过whl文件安装pytorch、torchvision以及torchaudio

第一步:进入whl文件所在文件夹

Anaconda安装Pytorch2.0.0(GPU版本)步骤_第2张图片

第二步:通过命令安装whl文件 

pytorch:

pip install torch-2.0.0+cu117-cp39-cp39-win_amd64.whl

torchvision:

pip install torchvision-0.15.0+cu117-cp39-cp39-win_amd64.whl

torchaudio:

pip install torchvision-0.15.0+cu117-cp39-cp39-win_amd64.whl

5. 验证

import torch
print(torch.__version__)    #显示版本
print(torch.cuda.is_available())

Anaconda安装Pytorch2.0.0(GPU版本)步骤_第3张图片

若输出为True,则安装成功!!! 

你可能感兴趣的:(python)