如何在openEuler-22.03-LTS版本上安装Pytorch?

运维
1

  1. 下载安装Miniconda3-latest-Linux-x86_64.sh，进行环境的安装配置.
    $sh ./Miniconda3-latest-Linux-x86_64.sh

  2. 使用如下命令安装pytorch.
    $conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia

  3. 测试pytorch是否安装成功.
    代码1：
    from future import print_function
    import torch
    x = torch.rand(5, 3)
    print(x)

    代码2：
    import torch
    torch.cuda.is_available()
    输出：True
    如果以上两段代码输出无异常，表明环境搭建成功。

2 个赞
2

经典好贴，点个赞