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
4. 通过whl文件安装pytorch、torchvision以及torchaudio
第一步:进入whl文件所在文件夹
第二步:通过命令安装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())
若输出为True,则安装成功!!!