PyTorch安装教程(Linux)

安装准备

  1. 请先安装Anaconda或者Miniconda并配置好清华源
  2. 输入命令nvidia-smi,查看CUDA版本,可以看到是10.1
    在这里插入图片描述

安装教程

  1. 用conda新建一个名为pytroch的环境
conda create -n pytorch python=3.7
  1. 激活刚才创建的名为pytorch的环境
conda activate pytorch
  1. 进入Pytorch官网CUDA版本选择之前我们查询的。然后复制下方的安装命令
    在这里插入图片描述
  2. 在第2步激活的pytorch环境中运行第3步复制的安装命令
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

验证安装

  1. 验证torch能不能正常使用,正常打印则说明安装成功
import torch
x = torch.rand(5, 3)
print(x)
  1. 再来看看CUDA是否能用,如果打印True则说明能用
import torch
print(torch.cuda.is_available())

引用参考

https://pytorch.org/get-started/locally/
  • 3
    点赞
  • 7
    收藏
    觉得还不错? 一键收藏
  • 打赏
    打赏
  • 2
    评论
安装PyTorch需要先确定Python版本和操作系统类型,以便选择正确的PyTorch版本。以下是在Windows、MacOS和Linux安装PyTorch的步骤: **在Windows上安装PyTorch** 1. 安装Python:下载并安装Python 3.x版本,建议使用Anaconda发行版,因为它包含了很多科学计算工具库。 2. 打开Anaconda Prompt:在开始菜单中搜索“Anaconda Prompt”,然后打开。 3. 安装PyTorch:在Anaconda Prompt中输入以下命令,根据需要选择CUDA版本和PyTorch版本: ``` conda install pytorch torchvision torchaudio cudatoolkit=xx.x -c pytorch ``` 其中,xx.x表示CUDA版本号,PyTorch版本号可以在官网上查看。 **在MacOS上安装PyTorch** 1. 安装Python:MacOS系统自带Python,也可以使用Homebrew或Anaconda安装Python。 2. 打开Terminal:在“应用程序”中找到“终端”并打开它。 3. 安装PyTorch:在Terminal中输入以下命令,根据需要选择CUDA版本和PyTorch版本: ``` pip3 install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html ``` 其中,cu111表示CUDA版本号,PyTorch版本号可以在官网上查看。 **在Linux安装PyTorch** 1. 安装Python:Linux系统一般自带Python,也可以使用包管理器安装Python。 2. 打开Terminal:打开终端窗口。 3. 安装PyTorch:在Terminal中输入以下命令,根据需要选择CUDA版本和PyTorch版本: ``` pip3 install torch torchvision torchaudio ``` 如果需要使用GPU加速,需要先安装相应的CUDA和cuDNN库,并在安装PyTorch时指定CUDA版本号。 以上是在各个操作系统上安装PyTorch的基本步骤,具体安装过程可能会有些许差异。建议先查看官方文档并根据自己的需求进行安装

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

Xavier Jiezou

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值