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

  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 Likes

经典好贴,点个赞