第一次用,记录一下
先把自己的代码打包上传到jupyter的目录中,并解压
unzip XXX.zip
我自己需要的python版本并没有,所以要自己创建一个虚拟环境
conda create -n my-env python=自己需要的版本号
//更新bashrc中的环境变量
conda init bash && source /root/.bashrc
//切换到创建的虚拟环境------my-env
conda activate my-env
然后配置pytorch环境,注意安装顺序,版本看自己需要
//首先安装 PyTorch 库 torch 和 torchvision。
pip install torch==2.1.1 torchvision==0.16.1
//然后安装 torch-geometric。
pip install torch-geometric==2.5.0
//最后安装 torch-scatter。
pip install torch-scatter==2.1.2
安装torch-scatter的时候报错了,错误如下:
note: This error originates from a subprocess, and is likely not a problem with pip.
ERROR: Failed building wheel for torch-scatter
Running setup.py clean for torch-scatter
Failed to build torch-scatter
ERROR: Could not build wheels for torch-scatter, which is required to install pyproject.toml-based projects
这些版本在我的pycharm上都可以正常安装,不知道什么原因。
最后使用下面这个指令安装成功了(后面的网址需要去torch-geometric网站看自己的torch版本对应的那些)
pip install --no-index torch-scatter -f https://pytorch-geometric.com/whl/torch-2.1.1%2Bcpu.html
有requirements.txt就直接安装这个,没有就一个一个自己安装
已经可以跑了!开心