最新消息:雨落星辰是一个专注网站SEO优化、网站SEO诊断、搜索引擎研究、网络营销推广、网站策划运营及站长类的自媒体原创博客

Why is Pytorch Cuda not detecting my GPU? - Stack Overflow

programmeradmin3浏览0评论

Here's a slice of my code:

import torch
print(torch.cuda.is_available())
print("CUDA Version:", torch.version.cuda)

Here's the output:

False
CUDA Version: 12.6

I looked at task manager, and it says I have a GPU. I tried installing Pytorch with GPU support using this Powershell command:

pip3 install torch torchvision torchaudio --index-url 
发布评论

评论列表(0)

  1. 暂无评论