CUDA Nvidia GTX 1650 |蟒蛇

问题描述 投票:0回答:1

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

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

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

NVIDIA toolkit version/information

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

  1. 无法从pytorch获得任何输出
import torch
torch.cuda.device_count()
torch.cuda.is_available()

Here is the output

  1. OpenCV 尝试通过 CMAKE 手动安装以及预制的 pip conda install
import cv2
print(cv2.cuda.getCudaEnabledDeviceCount())

在 openCV 中我得到输出 0

我尝试过什么:

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

python opencv pytorch cuda pycuda
1个回答
0
投票

无法从pytorch获得任何输出

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

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

如果没有成功,请尝试在 conda shell 中运行以下命令,然后尝试重新运行第一个脚本:

conda install -c pytorch -c nvidia -c conda-forge pytorch torchvision pytorch-cuda
© www.soinside.com 2019 - 2024. All rights reserved.