[今日一水] 安装Pytorch-GPU基本操作是对的，但为啥GPU就是不能用？

氵文大师

已于 2024-02-18 17:05:07 修改

阅读量476

点赞数 1

分类专栏：每日一氵文章标签： pytorch 深度学习

于 2021-09-05 20:04:50 首次发布

本文链接：https://blog.csdn.net/HaoZiHuang/article/details/120119868

版权

157 篇文章 11 订阅

订阅专栏

这都几年了，重装Ubuntu环境，与NVIDIA显卡驱动斗争，安装Pytorch——总之就是配环境上，也算相当熟练了，然后win上安装Torch死活用不了GPU??? 离天下之大谱

应该是是我记忆错乱了？我在Ubuntu上配环境就是安装miniconda、CUDA、cuDNN，然后：

pip install torch -i https://mirror.baidu.com/pypi/simple
# 后边顺便安装个 torchvision / opencv 什么的

然后验证一下torch.cuda.is_available() 是True就行

换了win上，最多更新一下NVIDIA显卡驱动，然后操作一样，他就不行了…

自从我去年重装了笔记本环境(我攒了几年的环境啊!!!)，俺的win上基本就没用过GPU的Torch…

直到我看到了这个：

的这个截图：
在这里插入图片描述
淦！！！要这样安装(截图自上边那篇博客)

我顿时悟了，这不就是当年按照网上的教程，一步步来的么…

在这里插入图片描述

被这个问题折腾这么久，我他妈是个憨憨吧…

后记：
hxdm，一定要早点儿睡觉，经常熬夜的人，栈小，记忆容易溢出…

UPDATE:
2024.02.18
执行

import torch
print(torch.__version__)
print(torch.version.cuda)

如果 torch.version.cuda 返回 None, 说明你安装的torch是CPU版本的…
conda uninstall torch 卸载之后重装吧

离谱的是，我明明照着torch官方 conda 安装GPU版本的指南来的，咋还能只下载CPU版本…

关注