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