跳转到内容

欢迎来到Riguz的小站!这是一个私人wiki,用来记录一些我的笔记。

PyTorch get started

来自WHY42
Riguz​(留言 | 贡献)2023年12月11日 (一) 04:31的版本 (→Install pytorch)

Conda Installation

conda create --name deeplearning python=3.11
conda activate deeplearning
python --version
// 3.11.5

Install pytorch

conda install pytorch::pytorch torchvision torchaudio -c pytorch

To verify:

import torch
x = torch.rand(5, 3)
print(x)