一、安装显卡驱动最新版本
1、nvidia显卡驱动更新链接
官方驱动 | NVIDIAhttps://www.nvidia.cn/Download/index.aspx?lang=cn
2、查看显卡型号
打开任务管理器,找到nvidia的型号,这里是NVIDIA GeForce MX150
3、选择要下载的驱动类型
对照自己电脑的显卡类型选择驱动类型
下载类型默认即可
4、下载安装
安装路径可以任选,其他全部默认即可
二、确定cuda driver的版本
打开命令窗口,Windows+r,输入nvidia-smi,查看cuda driver版本号,这里是12.2
三、确定cuda runtime版本,并下载pytorch
1、打开pytorch官网
2、选择版本
选择比12.2小的版本即可,这里选择cuda 11.8
如果需要更小版本而选项中没有提供,可以下拉点击previous... 查看更早版本
另外ctrl+f可以实现浏览器页面的文字搜索,很好用的功能,可以帮助我们实现对所需版本的查找。
如果安装了anaconda即选择conda包,也可选择pip,但conda优先。
3、创建一个虚拟环境,存放pytorch
创建虚拟环境:conda create -n 环境名称 python=3.11
4、复制命令进行下载(cudnn不需要单独安装,在pytorch中有集成)
如果下载过慢,建议使用镜像,使用镜像需要选用pip,复制改代码,并加上一行:
-i https://pypi.tuna.tsinghua.edu.cn/simple some-package
这样是错的!!!镜像只能下载cpu的torch!!
参考下面这个博客(10条消息) torch.cuda.is_available()返回false——解决办法_Nefu_lyh的博客-CSDN博客https://blog.csdn.net/qq_46126258/article/details/112708781(1)删除原来的cpu版本的torch、torchvision、torchaudio,地址如下:
删除内容如下:
(2)直接在网站中下载对应的torch、torchvision、torchaudio版本
网址:
download.pytorch.org/whl/torch_stable.htmlhttps://download.pytorch.org/whl/torch_stable.html注意win_amd64等于win x86_64
(3)用cmd打开下载的路径,本别执行:
pip install +名字,先下载torch那个,名字要加后缀
四、检测安装结果
依次输入这三句
python
import torch
torch.__version__
torch.cuda.is_available()
输出torch版本号,证明安装好了
如果输出TURE就证明GPU可以用了
当当!!