抱歉,您的浏览器无法访问本站
本页面需要浏览器支持(启用)JavaScript
了解详情 >

安装GeForce驱动程序

下载

英伟达官网
手动搜索驱动程序:
产品类型:GeForce
产品系列:是什么选什么,没有就找最接近的(后面还有产品进一步找你的显卡),Notebooks表示笔记本,一般选这个。
产品、操作系统、语言选自己的
下载类型:选Studio就好,没有就选全部或者Game Ready都可以
最后开始搜索
搜索结果一般可以随便选,选最新的或者比较新的都可以。

安装

注意!安装之前关闭火绒!
用管理员身份运行
可以自定义,安装图形驱动程序就好,PhysX系统软件一般不用安,Experience不用(安也没事),可以执行清洁安装(反正也没什么配置不是,又不玩游戏)
等待安装成功

安装CUDA Toolkit

打开cmd(管理员最好),输入

1
nvidia-smi

出来个框框就说明上一步驱动程序安装成功。

下载

在框框的右上角有个“CUDA Version:”,记作【CUDA Version】比如12.2,就是你能安装的CUDA Toolkit的最高版本,可以就安那个版本,也可以稍低一点。不用担心和你的pytorch所需的CUDA不同,可以向下兼容的,你下pytorch的时候下所需版本完全没问题;反而,如果你刚好安装你所需版本的CUDA Toolkit,反而可能会有问题,我就有这样的经历。
在下面这个网址下载,把12-2改成你的【CUDA Version】就好,注意是-不是.

1
https://developer.nvidia.com/cuda-12-2-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

选择你的配置(Windows,10,exe(local))
点击Download就好,很快的。

安装

用管理员身份运行
安装同上,Experience也可以不安
等待安装成功
打开cmd(管理员最好),输入

1
nvcc -V

能看到版本之类的信息。

安装Pytorch

进入网址
红框框那,选择Stable,Windows,pip(注意要点pip),python,如果有你所需的CUDA版本就点他,没有就选稍高些的版本,或者点击这个链接进入之前版本下载(Ctrl F 查找,比如版本11.6就搜116,注意要是pip)。
找到链接复制下来

1
2
3
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117(框框那)
或者
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116(之前版本那)

进入要安装Pytorch的虚拟环境,在上面的链接后面加你的系统代理,”–proxy http://127.0.0.1:1080“ ,特别注意系统代理的端口,(用pip安装其他包也可以这么做)

1
pip install torch==1.13.1+cu116 torchvision==0.14.1+cu116 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu116 --proxy http://127.0.0.1:1080

验证

1
2
3
python
import torch
torch.cuda.is_available()

返回True,说明成功


修曦所著,转载注明
如果能帮到人就更好啦

评论