1 NVIDIA驱动安装
一般来说,驱动可以使用兼容的最新版本
window安装
https://www.nvidia.cn/geforce/drivers/
2 CUDA Toolkit安装
(1)CUDA Toolkit版本要求
win+R输入nvidia-smi 查询可以安装CUDA Toolkit版本,CUDA Toolkit版本小一点没有关系
(2)下载CUDA Toolkit并直接运行安装
https://developer.nvidia.com/cuda-12-1-0-downloads
(3)检查运行
nvcc-V
3 Pytroch安装
直接取官网选择合适的版本,在通过pip或conda安装即可
https://pytorch.org/