1. pytorch与python版本不对应
这是官网的安装命令,但还是报错了。原因在于这个环境是python3.7环境,而torch2.1.2至少需要python3.8以上,版本不匹配所以安装不了。解决办法是重新建python3.8的环境。
2. pytorch与cuda版本不对应
如果安装的pytorch版本与本地安装的cuda版本不对应,也会出现这种情况。这里特别说明一下如何正确查看windows下安装的cuda版本。
在网上查阅资料,最常见的有两种方法,先说结论,这两种方法都是不对的!
方法一:nvidia-smi
在cmd终端输入 nvidia-smi 这里显示的是显卡驱动版本555.97所能支持的最高cuda版本,并不是我们实际安装的cuda版本。
方法二:控制面板
这种方法一般来讲没问题,但我就遇到了问题...按照这个cuda版本安装pytorch时还是出问题,后面才发现自己办公电脑上安装的是cuda12.3版本,至于为什么会有冲突,无非是两个原因:(1). 这里显示的 CUDA 12.5.78 依然只是显卡驱动所支持的最高cuda版本;(2). 之前有人卸载重装了更低版本的cuda,但控制面板这里并没有动态更新。至于是哪个原因就不清楚了。
正确查看已安装cuda版本的方式
方法一:默认安装路径
方法二:环境变量
方法三:nvcc -V(需要配置好环境变量)
说白了,要想清楚自己电脑安装的什么版本的cuda,最根本的方法是寻找源头,也就是安装路径或者相应的环境变量,不然很有可能被误导。