1.Anaconda
1.1虚拟环境创建
2.Nvidia驱动
3.CUDA驱动安装
4.Pytorch安装
具体的步骤如上:可参考另一位博主的博客非常详细:
主要总结一下我在安装中出现的问题:
1.版本冲突问题
pytorch与python 版本不匹配:
初始我创建的conda环境python版本为3.12.3与其适配的cuda版本为11.8.但服务器端通过nviia-smi命令查看结果如下:
最高安装cuda版本为11.7,于是在pytorch找之前版本的pytorch安装。则出现了版本冲突。
解决方法:将python版本换为3.10即可。