附上文档开源地址:https://github.com/zergtant/pytorch-handbook
- Pytorch 简介
1.1 Torch 是一个与 Numpy 类似的张量(Tensor)操作库,与 Numpy 不同的是 Torch 对 GPU 支持的很好。
1.2 PyTorch 和 Torch 都使用的是相同的底层,只是使用了不同的上层包装语言。lua是Torch的上层包装语言,python是pytorch的上层包装语言。
1.3 PyTorch 是一个基于 Torch 的 Python 开源机器学习库,用于自然语言处理等应用程序。 它主要由 Facebook 的人工智能研究小组开发。
1.4 PyTorch 是一个 Python 包,提供两个高级功能:
-具有强大的 GPU 加速的张量计算(如 NumPy)
-包含自动求导系统的的深度神经网络 - Pytorch 环境搭建
anaconda 安装完成后可以开始创建环境,这里以 win10 系统为例。打开 Anaconda Prompt。
2.1 pytorch 为环境名,这里创建 python3.6 版。
–conda create -n pytorch python=3.6 ipykernel
#新建的环境是没有安装安装ipykernel的所以无法注册到Jupyter Notebook中,所以先要准备下环境,这里同时在创建环境时为其安装ipykernel,为后来jupyter使用本环境做铺垫,
2.2 安装torch和torchvision
–切换到 pytorch 环境:activate pytorch
–安装 GPU 版本pytorch,这里我通过cuda官网安装cuda10.1后,然后通过pytorch官网给的命令安装和cuda10.1配套的torch1.4.0和torchvision0.5.0,但网络太慢了。然后通过清华镜像安装,但镜像里没有这个版本的,只能在官网下载两者的whl包文件,然后使用pip本地安装whl文件,下载时可以通过迅雷下载,比浏览器自带下载器快,如下图。
2.3 测试cuda安装,torch安装是否成功
–win 下查看 cuda 版本命令:nvcc -V
–测试pytorch,则命令行进入创建的pytorch环境,输入python进入python命令行,然后import torch看是否报错!
遇到问题:
1、pytorch安装失败的解决过程
使用conda install pytorch -c pytorch命令安装pytorch遇到部分安装问题,没安装进度的包都是外网地址上的包。
使用conda list测试,一个都没安装上!
使用conda install pytorch,pytorch包的源地址换为清华源,但还是失败
最后通过官网下载和cuda10.1配套的torch1.4.0和torchvision0.5.0的whl包文件,然后使用pip本地安装whl文件,下载时可以通过迅雷下载,比浏览器自带下载器快。
安装成功,测试函数使用,一一通过测试!