mac安装pytorch
时间: 2023-05-24 08:05:18 浏览: 69
1. 打开终端,输入以下命令安装conda:
```
/usr/bin/ruby -e "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/master/install)"
```
2. 安装anaconda:
```
brew install anaconda
```
3. 创建一个conda环境并激活:
```
conda create --name pytorch_env python=3
```
```
source activate pytorch_env
```
4. 安装pytorch:
```
conda install pytorch torchvision -c pytorch
```
或者
```
conda install pytorch torchvision cpuonly -c pytorch
```
如果需要安装cuda版本,则需要先安装cuda,并使用对应的环境安装PyTorch。
5. 验证PyTorch是否正确安装。打开python并输入以下代码:
```
import torch
print(torch.__version__)
```
如果可以成功导入且输出了正确的版本号,则PyTorch已经安装成功。
注意:如果之前已经安装过pytorch,则需要先清除环境,并按照以上步骤重新安装。可以使用以下命令查看已有的环境:
```
conda info --envs
```
可以使用以下命令清除环境:
```
conda env remove --name pytorch_env
```
阅读全文