作者主页(文火冰糖的硅基工坊):文火冰糖(王文兵)的博客_文火冰糖的硅基工坊_CSDN博客
本文网址:[PyTroch系列-1]:PyTroch深度学习框架的详细安装过程_文火冰糖(王文兵)的博客-CSDN博客
目录
2.3 带GPU硬件平台的PyTroch的安装(需要大致10min)
1. PyTroch概述
PyTorch是Facebook推出的一个开源的Python深度学习框架。
PyTorch的前身是Torch,其底层和Torch框架一样,但是使用Python重新写了很多内容,不仅更加灵活,支持动态图,而且提供了Python接口。
2. PyTroch深度学习框架的详细安装过程
2.1 登录到PyTroch官网
根据自己的实际环境配置PyTroch,获取安装命令。
2.2 不带GPU硬件平台的PyTroch的安装
备注:选择此安装,在人工神经网络训练时,就不能选择GPU, 只能选择cpu,对于没有GPU的硬件平台,这样的安装可以节省硬盘空间。
conda install pytorch torchvision torchaudio cpuonly -c pytorch
2.3 带GPU硬件平台的PyTroch的安装(需要大致1小时)
备注:选择此安装,在人工神经网络训练时,可以选择cpu,也可选择GPU,对于有GPU的硬件平台,这样的选择后续根据的灵活,但需要更多的硬盘工具存放安装包。
创建pytorch环境:conda create -n pytorch1.8_py3.8
激活pytorch环境:conda activate pytorch1.8_py3.8
安装pytorch最新版本: conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
或
安装pytorch 指定版本: conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio=0.8.0 cudatoolkit=11.1 -c pytorch
2.4 检验安装
- 进入python解析器
(pytorch1.8_py3.8) PS C:\Users\Administrator> python Python 3.8.10 (default, May 19 2021, 13:12:57) [MSC v.1916 64 bit (AMD64)] :: Anaconda, Inc. on win32
Type "help", "copyright", "credits" or "license" for more information.
- 导入pytorch库
>>> import torch
- 调用pytorch库函数,检查GPU是否支持。
>>> torch.cuda.is_available()
False
- 退出python解析器,回到conda环境
>>> quit()
第3章 安装其他的常用库
conda install matplotlib