编程笔记

# PyTorch 概述

# 教程

# 安装

相关链接：

  1. PyTorch (opens new window)
  2. CUDA (opens new window)
  3. cuDNN (opens new window)
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/ (opens new window)

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))

# 推荐阅读