PyTorch 环境安装与验证

note 约 1 分钟

本文是系列第 1 篇,共 2 篇

查看系列目录(2 篇)
  1. 1. PyTorch 环境安装与验证
  2. 2. Tensor 基础操作

本文是「PyTorch 学习笔记」系列第 1 篇。

安装 Anaconda

# 创建独立环境,避免污染 base
conda create -n pytorch python=3.11 -y
conda activate pytorch

安装 PyTorch

根据显卡选择 CUDA 版本,去 pytorch.org 复制对应命令。

验证 GPU 可用

import torch

print(torch.__version__)
print(torch.cuda.is_available())  # True 即成功

评论