pytorch GPU版本安装 python windows

annanconda环境

创建虚拟环境?pytorch19_gpu

create -n pytorch19_gpu python=3.9

激活环境

conda activate pytorch19_gpu

查找CUDA版本是12.0，查找方式，win + r输入cmd进入命令行模式，输入nvidia-smi，如下，

查找如下pytorch版本，得到安装命令（测试使用conda命令无法成功安装，使用pip3命令进行安装，命令如下）

Start Locally | PyTorch

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

如果在虚拟环境cmd命令界面，输入python进行python编辑模式。输入下面语句，即可测试是否成功安装

import torch?
print(torch.cuda.is_available())

发现成功安装，如下，

此外查询torchvision，如果没有安装，安装即可，torch

在安装上述版本前，已经创建了pytorch19,python版本是3.9，安装命令如下

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

pytorch 2.1.2

安装其他库的命令，如下

pip install kornia==0.6.11

pip install matplotlib

pip install opencv-python

以上是LightGlue-main在windows下运行的环境配置介绍