PyTorch 安装和简单测试 for Mac

avatar 2024年03月19日14:53:58 0 282 views
博主分享免费Java教学视频,B站账号:Java刘哥 ,长期提供技术问题解决、项目定制:本站商品点此

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

 

一、访问官网,获取安装命令

访问地址:https://pytorch.org/get-started/locally

选择要安装的环境配置,复制命令

我的命令为:

conda install pytorch::pytorch torchvision torchaudio -c pytorch

 

二、在终端运行命令

如果没有安装 conda,请先安装,参考

1、新建虚拟环境

conda create -n pytorch2.2.1 python=3.7

2、激活虚拟环境

conda activate pytorch13

 

3、安装pytorch

我的Mac下命令是

conda install pytorch::pytorch torchvision torchaudio -c pytorch

我的Windows命令是(使用 nvidia-smi 命令查看CUDA版本,我的为11.6

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

 

4、进入python命令窗口

python

 

5、查看pytorch版本号

import torch
print(torch.__version__)

 

6、计算一个矩阵计算

N, D = 3, 4
x = torch.tensor(torch.rand(N,D))
y = torch.tensor(torch.rand(N,D))
z = torch.tensor(torch.rand(N,D))
a = x*y
b = a+z
c = torch.sum(b)
c

 

 

 

 

  • 微信
  • 交流学习,服务定制
  • weinxin
  • 个人淘宝
  • 店铺名:言曌博客咨询部

  • (部分商品未及时上架淘宝)
avatar

发表评论

avatar 登录者:匿名
匿名评论,评论回复后会有邮件通知

  

已通过评论:0   待审核评论数:0