pytorch安装教程
时间: 2023-08-28 13:18:59 浏览: 123
Pytorch安装教程
好的,以下是PyTorch的安装教程:
1. 首先需要安装Anaconda,可以在官网下载对应的安装包(https://www.anaconda.com/products/individual)。
2. 安装Anaconda后,在命令行中输入以下命令创建一个新的conda环境,并命名为“pytorch”:
```
conda create --name pytorch python=3.7
```
3. 激活刚创建的conda环境:
```
conda activate pytorch
```
4. 在conda环境中安装PyTorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
其中,cudatoolkit=10.2表示使用CUDA 10.2版本,如果你的机器上没有安装CUDA可以省略这个参数。
5. 安装完成后,在Python命令行中输入以下代码测试PyTorch是否安装成功:
```
import torch
x = torch.rand(5, 3)
print(x)
```
如果输出了一个5行3列的随机矩阵,则说明PyTorch安装成功了。
希望以上内容能够帮助你安装好PyTorch。
阅读全文