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虚拟环境并且对此进行重新安装。 安装正确