目录
首先我们需要先了解一下pytorch
PyTorch是一个开源的python机器学习库,基于Torch,用于自然语言处理等应用程序。它是一个基于Python的可续计算包,提供两个高级功能:
1、具有强大的GPU加速的张量计算(如NumPy)。
2、包含自动求导系统的深度神经网络。
以Python优先的深度学习框架,不仅能够实现强大的GPU加速,同时还支持动态神经网络。
安装教程(详细)
一. 安装 Anaconda
Anaconda 是一个用于科学计算的 Python 发行版,支持 Linux, Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包。
二.安装Pytorch
1、配置虚拟环境
1.1 下载完anaconda后,可以在应用栏里输入Anaconda Navigator
1.2 进入之后,点击launch进入anaconda prompt
1.3 使用清华镜像源网站
接着就是切换到国内的镜像源。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
将上面的四行代码依次输入到anaconda prompt里,这样就可以配置好了。
1.4 创建pytorch虚拟环境
在Anaconda Navigator中选择environment,点击create(如下)
创建一个pytorch的虚拟环境
我这边是已经创建过了,用的是上面这个办法。
还有另外一种方法,就是可以在命令行里直接输入
conda create -n pytorch python=3.8
具体python版本根据你实际情况来看。最后,在加载过程中会弹出提示,输入 y,就安装成功了。
1.5 检验环境是否安装成功
conda info --envs
在cmd命令行输入conda info --envs查看
如果出现pytorch环境,即为安装成功。
2、安装pytorch
配置好虚拟环境后,我们就可以开始下载pytorch了
2.1 进入pytorch官网:PyTorch
根据自己的系统版本,在寻找安装命令代码。
像我自己就是选择windows系统+cuda11.3(箭头指的那个三行就按上面的来)。
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
选择好之后复制最后一行的代码到命令行上去运行就可以了。
(如果用镜像源,记得把后面的-c pytorch删了在运行)
2.2 检验
打开命令行
输入python
继续输入
import torch
torch.cuda.is_available()
显示Ture代表安装成功了
import torch vision
print(torch.__version__)
print(torch.version.cuda)
可以继续输入上面代码查看当前cuda版本和pytorch版本
效果图就差不多这样了,安装完想着终于结束折磨了(狗头),如果顺利的话,成功的流程就差不多上面一般,但是总会出现各种问题,比如路径不对,环境没有激活各种情况,需要根据自己的实际情况去找问题咯。