-
进入pytorch官网。
https://pytorch.org/
找到适合自己的版本,CUDA、平台、Python等,然后会自动提示“Run this command” 命令指令进行安装。我的是Python3.5(Anaconda4.2)+CUDA8.0+windows10,当然如果电脑没有GPU,CUDA就选none。 -
conda 安装Pytorch包
Win+R进入windows运行,输入cmd进入命令行窗口,输入以上命令稍等片刻可以自动下载安装了,如下图所示。
输入run this command对应的命令
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
安装成功。
验证代码:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
注意:若笔记本有GPU,要事先安装cuda,并打开 terminal,输入命令行 nvcc -V 查看cuda的版本号,详见参考链接。
参考文章一:《win10系统安装cudn, cudnn,配置深度学习环境》
https://blog.csdn.net/weixin_38410551/article/details/103933108
参考文章二:《GPU个人电脑环境运行配置指引》
https://download.csdn.net/download/weixin_38410551/12096918
参考链接:
https://blog.csdn.net/cuixing001/article/details/81952116
https://blog.csdn.net/weixin_38410551/article/details/103933108
https://download.csdn.net/download/weixin_38410551/12096918
.