前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >Pytorch环境安装准备

Pytorch环境安装准备

作者头像
是小张啊喂
发布2022-08-18 17:33:07
3010
发布2022-08-18 17:33:07
举报
文章被收录于专栏:软件软件

安装 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
本文参与 腾讯云自媒体分享计划,分享自作者个人站点/博客。
原始发表:2022-07-15,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档