1.完全删除以前的环境conda env remove -n pytorch
2.创建新的虚拟环境conda create -n pytorch python=3.7
3.进入pytorch官网找到自己需要的对应版本安装指令,如下图
输入指令开始安装pip3 install torch torchvision torchaudio
,这里需要注意的是我没有选择conda安装因为我试了很多次conda这个方法安装不上获取是镜像的问题,没有配置好好的源,但是pip这种方法也是可以用的并且配出来是可以跑的
4.检查下是否配置成功
输入指令python
,import torch
,print(torch.__version__)
,print(torch.cuda.is_available())
,如下图出现True为成功配置!
5.再根据个人需要安装一些其他的numpy,pandas等包即可!
6.这里记录下torch里的一些基本的torch-geometric等安装,下载torch_scatter,torch-sparse,torch-cluster,torch-spline-conv包,这些包依赖关系非常严重,下错了就是满屏红字。可以从中下载https://pytorch-geometric.com/whl/,找到对应的torch版本,例如从上面我们可以看到我们torch版本是1.13.1+cu117,则进入对应的目录,下载对应的python版本需要的几个文件,如下图,把文件移到服务器,进行pin install +文件名即可安装
最后直接pip install torch-geometric
,应该可以成功安装!
pytorch去官网安装不行了,换用下面这个方法
pip install torch==1.12.1+cu116 torchvision==0.13.1+cu116 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu116