Pytorch 利用 CUDA Nvidia GTX 1650

Question

我正在尝试做一些基于人工智能的Python工作，并希望通过cuda将压力从CPU转移到GPU，这样我的程序就会运行得更流畅。

系统： NVIDIA GTX 1650 笔记本电脑 I5-9300H WINDWONS 10 主页 22H2

可执行文件： 迷你康达3 Microsoft Visual studio 2022（甚至尝试过基本的 VS 代码）

但是在这里我遇到了奇怪的问题：

1. 无法从pytorch获得任何输出

import torch
torch.cuda.device_count()
torch.cuda.is_available()

2. OpenCV 尝试通过 CMAKE 手动安装以及预制的 pip conda install

import cv2
print(cv2.cuda.getCudaEnabledDeviceCount())

在 openCV 中我得到输出 0

我尝试过什么：

尝试从 vs code 更改为 microsoft vs code 2022 仍然相同 尝试更新 conda 中的所有内容： conda update --all

Answer 1

无法从pytorch获得任何输出

您没有在代码中将结果打印到控制台。 你可以尝试：

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

如果 cuda 在 torch 中不可用，请尝试在 conda shell 中运行以下命令，然后尝试重新运行第一个脚本：

conda install -c pytorch -c nvidia -c conda-forge pytorch torchvision pytorch-cuda