既然你搜到这个解决方案,就不需要解释google colab是什么,
首先安装Anaconda的环境,非常容易
!wget -c https://repo.continuum.io/archive/Anaconda3-5.1.0-Linux-x86_64.sh
!chmod +x Anaconda3-5.1.0-Linux-x86_64.sh
!bash ./Anaconda3-5.1.0-Linux-x86_64.sh -b -f -p /usr/local
import sys
sys.path.append('/usr/local/lib/python3.6/site-packages/')
我装的python是3.6版本
然后查看你的torch的版本
import torch
torch.__version__
我的版本是:1.10.0+cu111
接下来安装其他软件包,我需要安装pytorch_geometric and rdkit
执行命令
!conda create -n geometric python=3
!conda activate geometric
!conda install -y -c conda-forge rdkit
!conda install pytorch torchvision cudatoolkit -c pytorch
注意:在google上安装软件包都需要在前面加上!符号,英文的
继续安装
!pip install torch_sparse
!pip install torch_scatter
!pip install torch-cluster
!pip install torch-spline-conv
!pip install torch-geometric
环境配置差不多就可以运行了
!python create_data.py
安装cuda
!sudo lsb_release -a