1.打开官网 https://pytorch.org/
根据官网提示选择合适的命令在命令行中输入
Windows10版本,Pip安装,无CUDA,则输入:pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
下载大概率失败,直接下载上述网站的内容
进行安装 pip install 路径名
或者运气好可以直接成功安装:
2. 测试安装
输入python
输入代码:
-
from __future__ import print_function
-
import torch
-
x=torch.rand(5,3)
-
print(x)
3.配置Jupyter Notebook
安装已经安装过了,打开cmd,输入 jupyter notebook即可运行