vs2017 开始自己的第一个PyTorch程序

华为云博客 技术博客 17076 25

这是针对于博客vs2017安装和使用教程（详细）的PyTorch项目新建示例

博主还提供了其他几篇博客供大家享用：

VGG16处理cifar-10数据集的PyTorch实现

PyTorch 入门实战（五）——2013kaggle比赛 猫狗大战的实现

目录

一、安装PyTorch环境

二、新建项目

三、运行代码

一、安装PyTorch环境

1.确保自己的电脑里已经安装了CUDA和cuDNN等环境，安装步骤和安装tensorflow环境基本一致，可参看博主早期的一个安装教程：windows10 64位 Tensorflow安装--CUDA 9.1+cuDNN7.1.1+python3.6.4+tf1.6+vs2017，CUDA版本建议9.0，cudnn版本建议7.0到7.2。

2.确保自己的电脑里已经安装了Anaconda3环境，没有可以参看博主文章：Anaconda安装，可以直接网上下载：Anaconda官网

3.官网下载：

英文官网：PyTorch官网

中文官网：PyTorch中文官网

4.英文官网来到这个部分，运行提示的命令即可，torchvision可以是cuda80或者cuda90：

       

5.一般官网的下载速度会很慢，这里博主提供一个下载地址：Pytorch-0.4.1兼容CUDA9.0，下载之后到下载该文件的目录直接打开cmd，运行以下指令完成安装：

pip install torch-0.4.1-cp36-cp36m-win_amd64.whl

二、新建项目

1.和本博主的vs2017 开始自己的第一个Python程序这个方法一样新建一个项目，博主这里是PyTorch_sample：

                                                             

2.然后查看python环境看看是不是自己之前安装的anaconda环境，之后我们输入代码：


  
    
   
  1. 
    
    
     
    
    
    
    
    
     
    
      import torch
     
    
    
    2. 
   
  2. 
    
    
     
    
    
    
    
    
     
    
      x = torch.rand(1,9)
     
    
    
    3. 
   
  3. 
    
    
     
    
    
    
    
    
     
    
      print(x)
     
    
    
    4.

三、运行代码

1.我们点击附加...按钮，运行代码：

                                                  

2.可以看到输出结果：

说明我们可以用vs2017进行深度学习的PyTorch框架下的开发

                    

返回至原博客：vs2017安装和使用教程（详细）

文章来源: nickhuang1996.blog.csdn.net，作者：悲恋花丶无心之人，版权归原作者所有，如需转载，请联系作者。

原文链接：nickhuang1996.blog.csdn.net/article/details/86226478

（完）