Pytorch-GPU+PyG极简安装!!

本文提供了一种简单的方法在Anaconda环境中安装Pytorch-GPU和PyG。首先,检查CUDA版本,根据版本从Pytorch官网获取安装命令。通过pip指令安装Pytorch,并在安装完成后进行测试以验证安装成功。接着,按照类似步骤安装PyG,完成图神经网络库的配置。
摘要由CSDN通过智能技术生成

目录

一、安装pytorch

二、安装PyG


继极简安装TensorFlow-GPU版本之后,这边继续更新一个极简安装pytorch-gpu教程,依然在Anaconda下面安装,步骤如下:

一、安装pytorch

0、如果没有的话,先去安装个Anaconda、创建虚拟环境

1、打开英伟达控制面板,看看自己的GPU需要什么版本的CUDA,操作如下:

打开控制面板——》NVIDIA控制面板——》系统信息——》组件

比如我的,如图所示,我的版本是CUDA11.0

2、打开pytorch的官网,查看下相应的安装指令

官网地址:https://pytorch.org/get-started/locally/

打开之后如图所示,package部分建议选pip指令安装,感觉下载速度比较快;CUDA部分选上一步中所查到的版本,我的是11.0;然后把那个语句复制一下,我的是:

pip install torch===1.7.1+cu110 torchvision===0.8.2+cu110 torchaudi
  • 3
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值