PyTorch与PyTorch Geometric的安装过程
PyTorch与PyTorch Geometric的安装
GPU与CUDA,Python,PyTorch的匹配
1. 查看Linux系统中GPU的基础信息/NVIDIA Driver Version
nvidia-smi
nvidia-smi是nvidia 的系统管理界面 ，其中smi是System management interface的缩写，它可以收集各种级别的信息，查看显存使用情况。此外, 可以启用和禁用 GPU 配置选项 (如 ECC 内存功能)。
系统的Nvidia Driver Version决定着系统最高可以支持什么版本的cuda和cudatoolkit，Nvidia Driver是向下兼容的，详情如下（见Table 3. CUDA Toolkit and Corresponding Driver Versions）：
2. 查看当前CUDA版本：
cat /usr/local/cuda/version.txt
我自己的环境最高可支持10.1版本的cuda和cudatoolkit，当前是10.0，版本向下兼容，并无什么问题。
3. CUDA Toolkit匹配PyTorch
CUDA Toolkit 和PyTorch的对应关系（见官网）
！ 注意事项：服务器本身的CUDA版本与虚拟环境中安装的cudatoolkit包没有太大关系，一般安装pytorch时需要考虑的cuda版本指的应该是虚拟环境中安装的cudatoolkit包的版本
由于我需要用到v1.6.0的Pytorch，因此自己在虚拟环境里安装v10.1的CUDA Toolkit，系统GPU可接受最高版本v10.1。
torch与torchvision对应关系如下（详情见PyTorch / Vision）：
因此，我需要安装的如下：
CUDA Toolkit == 10.1
Python == 3.7
PyTorch == 1.6
安装PyTorch
pip install torch==1.6.0+cu101 torchvision==0.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
安装完成后可通过以下命令检查torch版本及对应的CUDA版本：
python -c "import torch; print(torch.__version__)" python -c "import torch; print(torch.version.cuda)"
再通过以下命令查看GPU是否可用：
python >>> import torch >>> torch.cuda.is_available() # GPU是否可用 >>> torch.cuda.device_count() # GPU数量 >>> torch.cuda.current_device() # 当前GPU >>> exit()
注意，GPU devices从0开始编号。
安装PyTorch Geometric
1. 快速安装
根据官网，如果PyTorch版本≥1.8.0，可以快速下载：
2. 自定义安装
自定义下载需要根据当前的PyTorch版本和CUDA版本下载相关的依赖，下载命令如下：
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-${TORCH}+${CUDA}.html
pip install torch-geometric其中， ${TORCH}替换为当前环境下的PyTorch版本，目前支持1.4.0、1.5.0、1.6.0、1.7.0、1.7.1、1.8.0、1.8.1、和1.9.0； ${CUDA}替换为指定的CUDA版本，目前支持cpu、cu92、cu101、cu102、cu110和cu111。
例如对于PyTorch 1.6.0和CUDA 10.1：
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html pip install torch-geometric
3. 版本依赖
使用自定义安装时，依然可能会出现安装失败的问题，因为pytorch_geometric几个相关库之间有比较强的依赖关系，建议是在自定义安装的基础上指定对应库的版本，例如对于pytorch1.6.0和cuda10.1：
pip install torch-scatter==2.0.5 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html pip install torch-cluster==1.5.8 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html pip install torch-sparse==0.6.7 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html pip install torch-spline-conv==1.2.0 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html pip install torch-geometric==1.6.1 -f https://pytorch-geometric.com/whl/torch-1.6.0+cu101.html
到此这篇关于PyTorch与PyTorch Geometric的安装的文章就介绍到这了,更多相关PyTorch Geometric安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要为大家详细介绍了如何使用Python编写电脑定时关机小程序,文中的示例代码讲解详细,感兴趣的小伙伴可以跟随小编一起学习一下2024-01-01
python pandas创建多层索引MultiIndex的6种方式
这篇文章主要为大家介绍了python pandas创建多层索引MultiIndex的6种方式，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2022-07-07
对pandas将dataframe中某列按照条件赋值的实例讲解
今天小编就为大家分享一篇对pandas将dataframe中某列按照条件赋值的实例讲解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-11-11
在数据分析过程中,使用pandas库进行数据清洗和过滤空值是基本而重要的步骤,首先,导入pandas库并加载数据,其次,利用isnull()或isna()方法检查数据中的空值,接着,可以使用dropna()方法过滤掉含有空值的行,或针对特定列过滤空值2024-09-09
这篇文章主要介绍了Pygame游戏开发中常用的图形绘制与键鼠事件实例，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习吧2022-11-11
Miniconda是一个轻量级、免费且开源的跨平台软件包管理系统,本文主要介绍了Linux上Miniconda的安装的实现步骤,具有一定的参考价值,感兴趣的可以了解一下2024-03-03
这篇文章主要介绍了python内置函数sorted()用法,结合实例形式较为深入的分析了Python内置函数sorted()功能、使用方法及相关操作注意事项,需要的朋友可以参考下2019-10-10
python中pymysql的executemany使用方式
这篇文章主要介绍了python中pymysql的executemany使用方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-01-01
这篇文章主要介绍了Django密码存储策略分析，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-01-01
今天小编就为大家分享一篇关于pip安装py_zipkin时提示的SSL问题对应，小编觉得内容挺不错的，现在分享给大家，具有很好的参考价值，需要的朋友一起跟随小编来看看吧2018-12-12
最新评论