1.Anaconda
? ? ? ? 1.1虛擬環境創建
2.Nvidia驅動
3.CUDA驅動安裝
4.Pytorch安裝
具體的步驟如上:可參考另一位博主的博客非常詳細:
Linux服務器配置Python+PyTorch+CUDA深度學習環境_linux cuda環境配置-CSDN博客https://blog.csdn.net/NSJim/article/details/115386936
主要總結一下我在安裝中出現的問題:
1.版本沖突問題
pytorch與python 版本不匹配:
初始我創建的conda環境python版本為3.12.3與其適配的cuda版本為11.8.但服務器端通過nviia-smi命令查看結果如下:
最高安裝cuda版本為11.7,于是在pytorch找之前版本的pytorch安裝。則出現了版本沖突。
解決方法:將python版本換為3.10即可。