文章目录
前言
记录一下近日下载pytorch时遇到的坑
cuda
查看当前可支持的最高cuda版本
快捷键
win+R
cmd
NVIDIA-smi
在cmd命令行内输入NVIDIA-smi即可查看当前NVIDIA显卡的driver version(显卡驱动),和可安装的最高cuda版本(最好别太高,也别太低)
显卡驱动更新
显卡驱动太旧的话,要下载好新版本的驱动
笔者用的是以下软件更新的
下载cuda
下载前还要去看一下目前pytorch适配的最高cuda版本(很重要!)
https://pytorch.org/get-started/locally/
以pytorch网页上的cuda版本为主!
我就掉进了这个坑里
图里显示的cuda版本最高为12.1,但我之前却下了12.4版本的,最后搞了好久