[pytorch学习]1.pytorch ubuntu安装

看完了Deep Learning with Python,尝试了部分Keras的demo代码。

感觉Keras虽然容易上手,能够快速搭建出一个通用的模型,但是缺乏对底层的控制。

同时,在使用了自己编译的支持CUDA 10的tensorflow之后,总感觉有些不可控的bug出现;即使仅仅只是把demo代码改为自己的小工程,也有诸如load_model不能放在循环外(否则cudnn报错),第一次model.predict正确,但是循环做第二个predict就出错。在网上搜索了好几天,也没有找到问题的解答,怀疑是没有使用官方的支持CUDA 10版本。

所以,决定转向到pytorch上。原因有三:1. 有官方的CUDA10正式版本,并且从网上的反馈来看,版本维护及支持比keras好;2. 能够对于深度学习的底层过程有更多的了解和控制;3. pytorch的嵌入式系统可移植性也很高,貌似速度更快。

系统: Ubuntu 18.04

硬件:1080Ti

1. Pytorch ubuntu安装

官方网站:https://pytorch.org/,选择正确的系统信息:

这里需要运行sudo命令,否则直接运行pip3会报错没有权限

sudo -H pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.0-cp36-cp36m-linux_x86_64.whl
sudo -H pip3 install torchvision

2. 测试安装是否正常

返回一个5X3的随机矩阵

3. 检查是否有CUDA支持

返回True

原文地址:https://www.cnblogs.com/westmole-site/p/10314663.html

时间: 2024-10-05 23:58:24

[pytorch学习]1.pytorch ubuntu安装的相关文章

一线开发者在Reddit上讨论深度学习框架:PyTorch和TensorFlow到底哪个更好?

本文标签:   机器学习 TensorFlow Google深度学习框架 分布式机器学习 PyTorch   近日,Reddit用户 cjmcmurtrie 发了一个主题为「PyTorch vs. TensorFlow」的讨论帖,想要了解这两大流行的框架之间各自有什么优势. 原帖地址:https://redd.it/5w3q74 帖子一楼写道: 我还没有从 Torch7 迁移到 TensorFlow.我玩过 TensorFlow,但我发现 Torch7 更加直观(也许是我玩得不够?).我也尝试了

Pytorch学习--编程实战:猫和狗二分类

Pytorch学习系列(一)至(四)均摘自<深度学习框架PyTorch入门与实践>陈云 目录: 1.程序的主要功能 2.文件组织架构 3. 关于`__init__.py` 4.数据处理 5.模型定义 6.工具函数 7.配置文件 8.main.py 9.使用 1.程序的主要功能: 模型定义    数据加载    训练和测试 2.文件组织架构: ```├── checkpoints/├── data/│   ├── __init__.py│   ├── dataset.py│   └── get_

wgan pytorch,pyvision, py-faster-rcnn等的安装使用

因为最近在读gan的相关工作,wgan的工作不得不赞.于是直接去跑了一下wgan的代码. 原作者的wgan是在lsun上测试的,而且是基于pytorch和pyvision的,于是要装,但是由于我们一直用的是python 2.7,所以无法从WGAN的页面上跳到pytorch的官方页面下载安装,需要安装github上的版本,这个是不需要python3.5的.重新装的系统,就安装了tensorflow和caffe,编译都没问题.但是当我安装pytorch和pyvision时,涉及到lmdb,过程出了问

Linux学习笔记——第一篇——Ubuntu安装与操作

笔者是Windows的使用者,由于Coding的需要以及在Linux下开发的方便,所以开始使用Linux. 当然笔者还是割舍不下Windows的,毕竟很多通讯工具等软件以及游戏在Linux下是没有发行的,所以笔者使用了虚拟机啊. 下面给出简单的安装过程. 1.下载虚拟机软件,笔者比较喜欢VMPlayer,因为它比较轻便且免费,并且很好的支持了拖拽复制功能(VM TOOL),当然也可以使用如VirtualBox.VPC等 链接:https://my.vmware.com/web/vmware/fr

重新学习Ubuntu -- 安装Git软件

好长时间没有来写相关的文章了,今天重新来学习--重新学习Ubuntu 安装Git并配置git 安装Git 在终端输入命令: sudo apt install git 显示的提示: 正在读取软件包列表... 完成正在分析软件包的依赖关系树 正在读取状态信息... 完成 将会同时安装下列软件:git-man liberror-perl建议安装:git-daemon-run | git-daemon-sysvinit git-doc git-el git-email git-gui gitkgitwe

分享《深度学习之PyTorch实战计算机视觉》PDF

下载:https://pan.baidu.com/s/1Yhf2Pv0gy63VgiMQTglgXA 更多资料:http://blog.51cto.com/3215120 <深度学习之PyTorch实战计算机视觉>两个版本的高清PDF,287页,带书签目录,文字可以复制. 如图: 原文地址:http://blog.51cto.com/3215120/2316431

分享《深度学习之Pytorch(廖星宇著)》+《PyTorch深度学习实战(侯宜军 著)》+源代码

下载:https://pan.baidu.com/s/1ewm1x3UeMe283PQVbDVIoA 更多资料分享:https://www.cnblogs.com/javapythonstudy/ <深度学习之Pytorch(廖星宇著)>PDF,232页,带书签,文字可以复制粘贴.<PyTorch深度学习实战(侯宜军 著)>PDF,121页,带书签,文字可以复制粘贴.配套源代码. 如图: 原文地址:https://www.cnblogs.com/javapythonstudy/p/

《深度学习与计算机视觉算法原理框架应用》PDF+《深度学习之PyTorch实战计算机视觉》PDF

下载:https://pan.baidu.com/s/12-s95JrHek82tLRk3UQO_w <深度学习与计算机视觉 算法原理.框架应用>PDF,带书签,347页.<大数据架构详解:从数据获取到深度学习>PDF,带书签,373页.配套源代码. 下载:https://pan.baidu.com/s/1P0-o29x0ZrXp8WotN7GzcA<深度学习之PyTorch实战计算机视觉> 更多分享:https://pan.baidu.com/s/1g4hv05UZ_

参考《深度学习之PyTorch实战计算机视觉》PDF

计算机视觉.自然语言处理和语音识别是目前深度学习领域很热门的三大应用方向. 计算机视觉学习,推荐阅读<深度学习之PyTorch实战计算机视觉>.学到人工智能的基础概念及Python 编程技能,掌握PyTorch 的使用方法,学到深度学习相关的理论知识,比如卷积神经网络.循环神经网络.自动编码器,等等.在掌握深度学习理论和编程技能之后,还会学到如何基于PyTorch 深度学习框架实战计算机视觉.<深度学习之PyTorch实战计算机视觉>中的大量实例在循序渐进地学习的同时,不断地获得成