zoukankan      html  css  js  c++  java
  • PyTorch安装

    从0开始在win10上安装PyTorch。

    PyTorch是一个类似Numpy的科学计算工具库,最大的特点是可以利用GPU进行并行计算。

    1.安装Anaconda 

     这里选Anaconda3 5.3.1  其集成的是python 3.7环境

     官网很慢,从清华镜像源下载,下载链接如下:

      https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Windows-x86_64.exe

    2.安装CUDA

    注意和python、pytorch版本对应,这里选择10.2  

    下载地址:

    https://developer.nvidia.com/cuda-10.2-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

    3.安装pytorch

      这里使用pip方式,conda方式安装速度过慢,镜像源好像都不能用了。 也要注意版本对应,在pytorch官网选择相应环境,自然会给出对应的安装语句。这里用的是豆瓣镜像源。

       cmd管理员权限运行,然后输入下列安装语句,大概一个G大小。

    pip install torch===1.6.0 torchvision===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple

    4.安装pycharm社区版

    https://www.jetbrains.com/pycharm/download/#section=windows

    5.测试是否成功

     pycharm新建工程,选择Anaconda下的python作为编译器。

    import torch
    
    print(torch.__version__)  # 检测pytorch版本
    print("gpu:", torch.cuda.is_available())  # 检测gpu是否可用
  • 相关阅读:
    ptunnel-简易使用
    socat-简易使用
    ncat-相关参数用法
    通过iodine简单实现dns隧道技术
    HTB-靶机-Safe
    HTB-靶机-Rope
    【mysql子查询&组合查询 05】
    【mysql 库表操作 07】
    【mysql插入&修改&删除 06】
    【mysql 连接查询 04】
  • 原文地址:https://www.cnblogs.com/peakz/p/13828605.html
Copyright © 2011-2022 走看看