1.cuda是否安装正确
nvcc -V
2.cudnn是否安装正确
nvidia-smi
3.安装Anaconda

4.更换清华源-不做描述
5.安装虚拟环境
6.安装pytorch
https://pytorch.org/get-started/locally/

7.出现问题排查:

我找了很久,前面的步骤都是正确的,但是一直都是安装不上torch,最后发现一直安装的都是cpu版本的,并未安装GPU版本,并且torch.cuda.is_available()一直都是False,所以,对python版本进行排查。发现是我创建的虚拟环境是3.6,所以我安装了一个py3.8虚拟环境并且对此进行重新安装。
安装正确


本文档记录了PyTorch的安装过程,包括检查CUDA和cuDNN的安装,安装Anaconda,创建虚拟环境,以及安装PyTorch。在安装过程中遇到的问题是始终无法安装GPU版本的PyTorch,原因是使用了Python 3.6的虚拟环境。解决办法是切换到Python 3.8环境并重新安装,最终成功安装并验证了torch.cuda.is_available()为True。
1829





