今天在安装环境时遇到报错:
The detected CUDA version (10.1) mismatches the version that was used to compile
PyTorch (11.8). Please make sure to use the same CUDA versions.
报错原因:安装的cuda版本不对应,我需要安装cuda的版本为11.8
下面开始cuda的安装:
1.打开网址CUDA Toolkit Archive | NVIDIA Developer
2.下载11.8版本就点击CUDA Toolkit 11.8,如下图:
按照指令依次执行即可