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下运行的环境配置介绍