Pytorch0.4.0环境搭建之详细教程、查看当前 Cuda和Cudnn版本

588 阅读1分钟
  • 🥇 版权: 本文由【墨理学AI】原创、各位大佬、一文读懂、敬请查阅
  • 🎉 声明: 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️

  • 📔 基础信息


    🟧 查看系统信息

    cat /etc/issue
    
    Ubuntu 16.04.7 LTS \n \l
    
    

    🟨 查看当前 Cuda 版本

    nvcc -V
    
    nvcc: NVIDIA (R) Cuda compiler driver
    Copyright (c) 2005-2019 NVIDIA Corporation
    Built on Sun_Jul_28_19:07:16_PDT_2019
    Cuda compilation tools, release 10.1, V10.1.243
    
    

    🟦 查看当前 Cudnn 版本

    cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
    
    
    #define CUDNN_MAJOR 7
    #define CUDNN_MINOR 6
    #define CUDNN_PATCHLEVEL 3
    --
    #define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
    
    #include "driver_types.h"
    
    

    📕 开始搭建 pytorch=0.4.0 环境


    🔴 第一步 Conda 创建环境 torch04

    conda create -n torch04 python=3.6.6
    

    默认就会安装如下库

    1-0

    🔵 第二步:安装 pytorch=0.4.0

    安装命令如下

    source activate torch04
    
    conda install pytorch=0.4.0 -c pytorch
    
    

    Conda 环境为适配 pytorch=0.4.0 会安装 Python 运行需要的 cudatoolkit 8.0

    1-1

    🟣 第三步:安装 其它依赖库

    pip install torchvision==0.1.6 matplotlib opencv-python tqdm scikit-image scipy==0.19.1
    

    📘 更多有效教程

    此次博文内容难以 以偏概全,如有不恰当的地方,欢迎评论区批评指正


    对于即将入行计算机视觉的小伙伴,墨理这里推荐收藏的干货博文目前如下

    5-0


    🚀🚀 墨理学AI


  • 🎉 作为全网 AI 领域 干货最多的博主之一,❤️ 不负光阴不负卿 ❤️
  • ❤️ 如果文章对你有帮助、点赞、评论鼓励博主的每一分认真创作
  • 橘色800.jpg