zoukankan      html  css  js  c++  java
  • 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/
    
  • 相关阅读:
    smarty基础
    smarty变量
    Google的Python代码格式化工具YAPF详解
    性能监控工具YourKit
    Scala学习笔记2
    Scala介绍
    线段树
    字符串分割方法
    Trie树
    Scala 学习笔记
  • 原文地址:https://www.cnblogs.com/ghgxj/p/14219129.html
Copyright © 2011-2022 走看看