医院网站主页面设计,网站组建,etw做的网站,网站团队人员目录
1.查看 torch 和 CUDA 版本
2.依次下载和 torch 和 CUDA 对应版本的四个依赖库pyg-lib、torch-scatter、torch-sparse、torch-cluster以及torch-spline-conv
3.下载并安装torch-geometric库 1.查看 torch 和 CUDA 版本
查看CUDA版本
nvcc -V
查看pytorch版本
pip s…目录
1.查看 torch 和 CUDA 版本
2.依次下载和 torch 和 CUDA 对应版本的四个依赖库pyg-lib、torch-scatter、torch-sparse、torch-cluster以及torch-spline-conv
3.下载并安装torch-geometric库 1.查看 torch 和 CUDA 版本
查看CUDA版本
nvcc -V
查看pytorch版本
pip show torch 2.依次下载和 torch 和 CUDA 对应版本的四个依赖库pyg-lib、torch-scatter、torch-sparse、torch-cluster以及torch-spline-conv 网址 https://data.pyg.org/whl/ 根据自己的torch版本和CUDA选择 选择需要的版本右键复制链接。 在命令行使用 wget 下载链接
wget -P 目标地址 https://data.pyg.org/whl/torch-2.0.0%2Bcu118/torch_sparse-0.6.18%2Bpt20cu118-cp310-cp310-linux_x86_64.whl
下载以上五个库的wheel文件之后使用 pip install 安装
pip install wheel文件地址
3.下载并安装torch-geometric库
网址PyG 根据自己的torch和cuda版本安装 torch-geometric库
pip install torch_geometric -f https://data.pyg.org/whl/torch-{torch版本号}cu{CUDA版本号}.html
# pip install torch_geometric -f https://data.pyg.org/whl/torch-2.0.0cu118.html
安装完成无报错
