PyTorch 概述

Tag: PyTorch-相关笔记 Posted on 2021-05-21 12:37:39 Edited on 2021-05-21 12:37:39 Views: 438

教程

安装

相关链接:

  1. PyTorch
  2. CUDA
  3. cuDNN
conda create --name torch
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch
pip install tensorflow

使用 pip 安装的时候必须注意 CUDA 版本,直接 pip install pytorch 安装的版本没有 CUDA 支持。 在这里找支持你的 CUDA 的 PyTorch 版本:https://pytorch.org/get-started/previous-versions/

pip install torch==1.7.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

验证安装是否成功(使用 IPython 的话要确保当前环境里有安装 IPython):

import torch
print(torch.__version__)
print(torch.version.cuda)
print(torch.cuda.get_device_name(0))

推荐阅读

未经允许,禁止转载,本文源站链接:https://iamazing.cn/