安装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 | pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117(框框那) |
进入要安装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 | python |
返回True,说明成功
修曦所著,转载注明
如果能帮到人就更好啦