PyTorch Install

少于 1 分钟读完

Install Pytorch,And some AI tools

Install Pytorch

Install Anaconda

下载

https://www.anaconda.com/download/

教程

https://docs.conda.io/projects/conda/en/latest/user-guide/tasks/manage-environments.html#

# check envs
conda info --envs
conda create --name pytorch
conda activate pytorch
# check cuda version in powershell
nvidia-smi

Install PyTorch

https://pytorch.org/get-started/locally/#windows-installation

conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

Install MNIST

conda install torchvision 

Install matplotlib

该工具可以通过tensor中存储的二维数据绘制图片

conda install matplotlib 

标签: ,

分类:

更新时间: