当前位置:   article > 正文

macOS M2使用conda配置pytorch环境_mac 安装conda

mac 安装conda

一、conda

首先我们需要有conda,miniconda的安装配置流程可以看这个Mac M1/M2芯片安装Miniconda并搭建tensorflow2环境

二、进入官网

选择对应的版本,复制下面的命令

conda install pytorch torchvision torchaudio -c pytorch

三、开始创建虚拟环境,下载pytorch

  • 进入终端,输入

  1. conda create -n torch-gpu python=3.9
  2. conda activate torch-gpu
  3. conda install pytorch torchvision torchaudio -c pytorch

第一条命令是创建名为torch-gpu,python版本为3.9的虚拟环境

第二条命令是激活环境

第三条命令就是安装pytorch

四、测试

  1. python
  2. import torch
  3. torch.__version__
  4. torch.device('mps')

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/620531
推荐阅读
相关标签
  

闽ICP备14008679号