macOS + PyTorch + Jupyter Notebook

1、安装 Jupyter Notebook:

pip3 install jupyter

2、安装 pyTorch :

https://pytorch.org 选择环境,会给出安装命令:

pip3 install torch torchvision

我的系统的显卡是 Intel 的集成显卡,没办法使用CUDA,故可以直接使用上面的pip安装。

如果是NVIDIA的显卡,可以参考链接 https://blog.csdn.net/junjun1996719/article/details/96426781 安装CUDA。

发布了69 篇原创文章 · 获赞 4 · 访问量 6万+

猜你喜欢

转载自blog.csdn.net/wangxiaosu/article/details/98375816