windows10上安装使用pytorch(pycharm也可以采用这种方法安装)

  1. 进入pytorch官网。
    https://pytorch.org/
    找到适合自己的版本,CUDA、平台、Python等,然后会自动提示“Run this command” 命令指令进行安装。我的是Python3.5(Anaconda4.2)+CUDA8.0+windows10,当然如果电脑没有GPU,CUDA就选none。

  2. 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
.

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章