pip 安装pytorch（gpu版本）匹配Python3.10

1.官网下载3.10.6Python版本

 

 

 直接双击exe文件 自定义安装

可修改路径，建议不要  

 2.官网cuda下载11.1（符合自己电脑版本）同时顺便下载cudnn

 

 

建议全选

3.镜像源（增加下载速度）

pip config set global.index-url http://pypi.tuna.tsinghua.edu.cn/simple

4.pip install pytorch（gpu）

方法一：

pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

方法2：本地下载

就是在官网上把安装包下载到本地路径在pip 下载

python -m pip install torch-1.13.0+cu117-cp310-cp310-win_amd64.whl
python -m pip install torchvision-0.14.0+cu117-cp310-cp310-win_amd64.whl
python -m pip install torchaudio-0.13.0+cu117-cp310-cp310-win_amd64.whl
 

5.验证

在命令行输入

Python

import torch
a = torch.cuda.is_available()

输出True就成功

在pycharm选中Python.exe环境输入

import torch
a = torch.cuda.is_available()
print(a)

ngpu= 1
# Decide which device we want to run on
device = torch.device("cuda:0" if (torch.cuda.is_available() and ngpu > 0) else "cpu")
print(device)
print(torch.cuda.get_device_name(0))
print(torch.rand(3,3).cuda()) 

查看结果