栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

MacBook M1配置Pytorch

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

MacBook M1配置Pytorch

提示:没有提示

文章目录
  • 前言
  • 一、Pytorch官网
  • 二、详细步骤
    • 1.查看Conda版本以及已经创建的虚拟环境
    • 2.创建一个新的虚拟环境,安装pytorch
    • 3.激活这个新的虚拟环境
    • 4.测试这个Pytorch可不可用
    • 5.重装Numpy
    • 6.重装Pytorch
    • 7.查看虚拟环境下的第三方库以及Numpy版本
    • 8.测试这个Pytorch
  • 总结


前言

本文用于MacBook M1配置Pytorch。


一、Pytorch官网

https://pytorch.org/get-started/locally/

二、详细步骤 1.查看Conda版本以及已经创建的虚拟环境

代码如下(示例):

conda --version
python --version
conda info --envs

2.创建一个新的虚拟环境,安装pytorch

代码如下(示例):

conda create -n pytorch

注意:中间有一个步骤需要输入“y”。


3.激活这个新的虚拟环境

代码如下(示例):

conda activate pytorch # 激活虚拟环境
conda install pytorch torchvision -c pytorch # 安装pytorch

4.测试这个Pytorch可不可用
import torch
x = torch.rand(5, 3)
print(x)

事实上这样会报错,因为Numpy的版本错误(Numpy版本太低)。
在接下来的步骤中,将卸载这个低版本的Numpy然后安装一个高版本的Numpy。

5.重装Numpy
conda uninstall numpy # 这样会移除刚刚安装的Pytorch以及一些其他的第三方库

现在查看已经安装过的第三方库

conda list

使用conda工具安装新版本的Numpy

conda install numpy

6.重装Pytorch
conda install pytorch torchvision -c pytorch

7.查看虚拟环境下的第三方库以及Numpy版本
conda list

8.测试这个Pytorch
import torch
x = torch.rand(5, 3)
print(x)

总结

提示:没有总结,哈哈哈。

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/989748.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号