2022-03-13 安装Pytorch(加避坑)

安装Pytorch的详细教程

    • 1、安装Anaconda/Miniconda
    • 2、在Anaconda中新建一个pytorch环境
    • 3、安装pytorch
    • 4、测试安装
    • 5、在pytorch中安装jupyter notebook
    • 6、深度学习机器视觉环境配置
    • 7、拓展:pytorch张量实验

1、安装Anaconda/Miniconda

  • 具体安装过程网上教程很多,可自行查看
    Anaconda/Miniconda环境安装
  • conda环境常用命令:
# conda create+创建一个名为python36的环境,制定版本是3.6
conda create --name python36 python=3.6

# sactivate 激活环境
activate python36

# 安装JupyterLab
pip install jupyterlab

# 启动JupyterLab
jupyter lab

# deactivate 退出当前环境,返回默认环境
conda deactivate

2、在Anaconda中新建一个pytorch环境

  1. 打开 Anaconda prompt 进入普通的base环境
# conda create + 环境名称 + 需要依赖的python3.7环境
conda create -n pytorch python=3.7
  1. 打开创建的这个环境
activate pytorch

3、安装pytorch

网上安装pytorch的教程有很多,有推荐速度较快的清华镜像源,也有推荐官网安装。本文仅展示官网安装方法,因为作者在安装了清华镜像后torch.cuda.is_available()返回false,说明cuda并没有安装成功,原因可能有插件不完备等等。

链接:Pytorch官网下载地址
2022-03-13 安装Pytorch(加避坑)_第1张图片
尽量选用stable稳定版本,我选择的是windows版本,pip安装(小坑:一般是用conda,但作者电脑的torch.cuda.is_available()总是返回false,换成pip安装后就变成了true),python语言,CUDA版本是11.3(版本要与电脑版本相对应)。

然后复制 run this command 上面的命令到自己新创建的pytorch环境中运行:

pip3 install torch==1.11.0+cu113 torchvision==0.12.0+cu113 torchaudio===0.11.0+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html

(小坑:如果用的是conda安装,最好不要随便去掉命令最后的-c pytorch,因为安装失败可能也与这个有关,等久一会儿就等一会儿,到时候重装还更麻烦费时。)

4、测试安装

在刚刚创建的pytorch环境命令行中输入 python 进入python。

测试>>>

import torch

如果没有报错说明安装成功。

再看看CUDA安装是否成功(很多人被卡住的地方)。

torch.cuda.is_available()

返回True则安装成功,大功告成,完美!
pytorch安装完成

5、在pytorch中安装jupyter notebook

jupyter是一个很好的交互界面,下面进行jupyter notebook安装教程。

  1. 进入pytorch环境(conda activate pytorch),输入以下代码,遇到选择就是yes:
conda install nb_conda
  1. 打开jupyter,直接在命令行运行jupyter notebook即可
    2022-03-13 安装Pytorch(加避坑)_第2张图片
    进入之后,选择这个pytorch环境即可。
    2022-03-13 安装Pytorch(加避坑)_第3张图片
    2022-03-13 安装Pytorch(加避坑)_第4张图片
    不报错说明成功!
    也可以用Pycharm进入创建的pytorch环境,配置简单可上网自查。

6、深度学习机器视觉环境配置

在完成以上pytorch环境的安装后,安装matplotlib、PIL和OpenCV三个python包。

进入pytorch环境

  • 安装matplotlib,输入以下命令:
conda install matplotlib

2022-03-13 安装Pytorch(加避坑)_第5张图片
matplotlib安装成功。

  • 安装PIL,输入以下命令:
conda install pillow

2022-03-13 安装Pytorch(加避坑)_第6张图片
PIL安装成功。
(ps:PIL仅支持到python2.7,PIL:Python Imaging Library是Python平台事实上的图像处理标准库,由于PIL仅支持到Python 2.7,加上年久失修,于是一群志愿者在PIL的基础上创建了兼容的版本,名字叫Pillow,支持最新Python 3.x,又加入了许多新特性,因此,我们可以直接使用conda install pillow安装PIL。)

  • 安装OpenCV,输入以下命令:
conda install opencv

2022-03-13 安装Pytorch(加避坑)_第7张图片
opencv安装成功。

7、拓展:pytorch张量实验

device的使用(成功说明GPU可用)
2022-03-13 安装Pytorch(加避坑)_第8张图片

参考链接:史上最成功安装Pytorch快速方法【亲测绝对有效,很好用很好用】

你可能感兴趣的:(专业学习,pytorch,深度学习)