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的安装相关推荐
- 【0基础入门Pytorch】Pytorch的简介与安装(Windows)
任务内容: Pytorch简介 Pytorch的安装步骤 1. Pytorch简介 PyTorch是一个基于Torch的Python开源机器学习库,用于自然语言处理等应用程序.它主要由Facebook ...
- 【Pytorch-从一团乱麻到入门】:1、Pytorch、jupyter notebook 的安装及设置
一. pytorch安装 1.查看CUDA版本 安装pytorch之前需要确定集群的CUDA版本,根据CUDACUDA版本去pytorch官网上选择合适的版本,根据提供的conda命令行安装即可 查看 ...
- pytorch的环境配置及安装(包括anaconda的安装)
目录 下载安装anaconda 如何验证是否已经安装成果 管理环境 查看计算机GPU型号是否支持coda 安装pytorch 查看驱动版本 进入pytorch首页PyTorch 检验安装是否成果 检查 ...
- Anaconda + Pycharm + Pytorch 一套太极组合拳安装教程
Anaconda + Pycharm + Pytorch 一套太极组合拳安装教程 1.下载以及安装Ananconda 1.1下载安装 1.2 Anaconda 环境配置 2.下载以及安装Pycharm ...
- Pytorch GPU版本whl文件安装
Pytorch GPU版本whl文件安装 安装pytorch的时候,用pip安装时网速实在太慢,换源也不太行,1.2G的文件,一个网络波动就开始疯狂红字.因此使用whl文件进行安装 用whl文件进行安 ...
- pytorch框架(1)安装那些破 ~ 事
前言:工欲善其事必先利其器 安装环境:Windows ,python3.7,需要安装pytorch 1.3.1版本 情况一.从简单的开始:安装cpu版本的pytorch 方式1:pip安装 (1)撞南 ...
- Pytorch Note1 Pytorch介绍
Pytorch Note1 Pytorch介绍 PyTorch的诞生 常见的深度学习框架简介 为什么选择PyTorch 总结一下Pytorch的特点 Python 优先 命令式体验 快速精益 安装 全 ...
- 1.0 深度学习回顾与PyTorch简介 - PyTorch学习笔记
P1 深度学习回顾与PyTorch简介 视频课程地址:点我 fly~~~ 本节课主要偏向于NLP,因为作者本人是做NLP方向NLP 预训练三种模型: BERT OpenAI GPT ELMo [NLP ...
- 【Pytorch】| Pytorch中softmax的dim的详细总结
[Pytorch]| Pytorch中softmax的dim的详细总结 关于softmax的理解 一维向量:dim=0和dim=-1结果相同,dim=1和dim=2会报错 二维张量:dim=1和dim ...
- [Pytorch框架] PyTorch 中文手册
PyTorch 中文手册 书籍介绍 这是一本开源的书籍,目标是帮助那些希望和使用PyTorch进行深度学习开发和研究的朋友快速入门. 由于本人水平有限,在写此教程的时候参考了一些网上的资料,在这里对他 ...
最新文章
- 高,低电平复位电路原理及加电容的作用
- 10分钟实现RPC框架
- spring设置全局异常处理器
- 微信小程序获取windowHeight出现不同页面高度不一致问题及解决方案
- 三元运算符(Java)
- 搜狗输入法精简_搜狗输入法10.10去图标精简版+9.4.21小米定制版
- 节约里程法解决VRP物流配送路径优化问题及局部优化改进-matlab代码
- java实现将汉字转为拼音并包含音调
- js模拟下雨js特效代码
- java导出建表语句_由 Java Entity 类自动生成 SQL建表语句
- 电脑检测工具eve_EVEREST中文版下载
- 2023年陕西师范大学宗教学考研上岸前辈备考经验指导
- 几何畸变的类型_几何畸变分为()和()。
- Prometheus自定义线程池指标暴露
- ros服务器打开网页变慢了,解决ros 在计算机没有限速情况下,有几个网站打开很慢,或者打不开。...
- Git Github操作简易教程
- 网吧免费上网的7种武器
- PPT基础(一)怎么把图片设置为背景
- Xilinx FPGA时钟及I/O接口规划(二)
- Altium Designer中的电路仿真