Pytorch环境安装准备

安装 anaconda

anaconda是一个开源的Python发行版本，其包含了conda、Python等180多个科学包及其依赖项。

官网地址：https://www.anaconda.com/products/distribution#windows

下载安装，验证是否安装成功

代码语言：javascript复制

$ conda --version
conda 4.12.0

// 依赖列表
$ conda list

安装Pytorch

官网地址：https://pytorch.org/

这里的计算平台可以选择简单一些入门，因为我的是A卡，所以直接选择CPU为计算平台的安装，如果是N卡，可以选择CUDA计算平台

CUDA下载地址：https://developer.nvidia.com/cuda-downloads

两种方式均可

代码语言：javascript复制

$ conda install pytorch torchvision torchaudio cpuonly -c pytorch

$ pip3 install torch torchvision torchaudio

pip和pip3都是python的依赖管理工具，只是版本不同

测试torch

代码语言：javascript复制

import torch

# torch version
print(torch.__version__)
# is support gpu
print('gpu:',torch.cuda.is_available())

1.11.0 cpu
gpu: False

打开文档

代码语言：javascript复制

jupyter notebook

https 网络安全 anaconda python pytorch

0 人点赞