1、 更新显卡驱动程序到最新版
上英伟达官网
https://www.nvidia.cn/drivers/lookup/
现在 ,更新!
2、查看显卡是否支持cuda。
打开显卡控制面板。nvidia
查看显卡的驱动程序版本,此处为566.36
上官网 https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
查看对应的下载 cuda的版本
如果都升级到了最新版,其实就是使用最高版本的cuda
3、下载 CUDA Toolkit
https://developer.nvidia.com/cuda-downloads
按照向导做。文件3G。建议用迅雷来下载。
下载完,按向导安装即可。文件较大,建议留好硬盘空间。
4、安装cudann
https://developer.nvidia.com/cudnn-downloads?target_os=Windows&target_arch=x86_64&target_version=10
5、验证安装
命令行两个命令
nvidia-smi
nvcc --version
显示数据了,说明安装正常了
6、安装pytorch
进入官网:https://pytorch.org/get-started/locally/
配置好对应的
在命令行运行下面命令
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
这是对应的pytorch指令。
注意:直接外网下载文件比较大,torch 就要2.5g
你可以先运行一下,然后把对应的文件的网址粘贴到 迅雷。这样会比较快而且可以断点续传。
下载完后, 使用 pip install torch-2.5.1+cu124-cp312-cp312-win_amd64.whl 进行安装。
以上内容为我自己的机器。