zoukankan      html  css  js  c++  java
  • Ubuntu系统Anaconda安装Pytorch,教你如何优雅的安装环境(1-5)

    一、明确前提

    首先我们安装的PyTorch是cuda版本的,所以首先要知道你的cuda版本(我的cuda是 10.2)

    二、打开pytorch官网,查看安装命令

    https://pytorch.org/get-started/locally/

    这里我没有安装最新版本怕不稳定

     选择一个适合你的版本(先别急输入命令哦!)

    a)官网安装

    conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 -c pytorch

    b)镜像安装(去掉 -c

    conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2 pytorch

    确定你已经配置过pytorch的国内镜像,如果使用国内镜像安装不成功就多试两次,总是不行可以选择官网安装(亲测!有时官网安装也不慢哦!)

    三、创建新的anaconda环境

    conda create -n python36torch15 python=3.6

    环境创建好之后,输入之前找到的安装pytorch的命令

    四、验证安装

    那你的cuda版本的pytorch安装好了没有呢?见证奇迹的时刻到了。

    在你创建的anaconda环境中(我的是python36torch15)输入

    python

    进入python代码编写环境

    输入代码

    #判断是否安装了cuda
    import torch
    print(torch.cuda.is_available())  #返回True则说明已经安装了cuda

    ——————  到这里pytorch就装好了开始你的pytorch之旅吧  ——————

    ******  下面这个是赠送的福利哦!相信你后面会用到的  ******

    #判断是否安装了cuDNN
    from torch.backends import  cudnn 
    print(cudnn.is_available())  #返回True则说明已经安装了cuDNN

    最后~祝科研顺利啊!!!

  • 相关阅读:
    word2010怎么把白色方框变成黑色方框?
    Ubuntu 14.04 安装 Sublime Text 3
    安装xmlspy之后,链接及邮箱等都用这个软件打开,怎样取消?
    SRAM、DRAM、SDRAM、DDR、DDR2、DDR3
    ROM和RAM区别
    shell脚本分析一
    重要网址
    vi/vim
    dump_stack使用
    BIOS、BootLoader、uboot对比
  • 原文地址:https://www.cnblogs.com/blogs-jch/p/13823721.html
Copyright © 2011-2022 走看看