autodl云端配置pytorch环境

第一次用,记录一下

先把自己的代码打包上传到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就直接安装这个,没有就一个一个自己安装

已经可以跑了!开心

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值