当前位置:   article > 正文

macOS M2使用conda配置pytorch环境 亲测有效果

macOS M2使用conda配置pytorch环境 亲测有效果

一、进入官网

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

pip3 install torch torchvision torchaudio

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

  • 1、conda create -n torch-gpu python=3.10
  • 2、conda activate torch-gpu
  • 3、conda install pytorch torchvision torchaudio -c pytorch

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

第二条命令是激活环境

第三条命令就是安装pytorch

三、测试

python
import torch
torch.__version__
import numpy as np
np_data = np.arange(6).reshape((2,3))
torch_data = torch.from_numpy((np_data))

 

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

闽ICP备14008679号