码迷,mamicode.com
首页 > Windows程序 > 详细

win10/windows 安装Pytorch

时间:2020-04-28 15:26:39      阅读:182      评论:0      收藏:0      [点我收藏+]

标签:index   targe   技术   data-   style   upload   imageview   cap   uda   

https://pytorch.org/get-started/locally/

去官网,选择你需要的版本。

技术图片

 

pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

命令行执行。

 

 CUDA(Compute Unified Device Architecture),是显卡厂商NVIDIA推出的运算平台。查看自己的CUDA版本步骤如下:

 

1.打开控制面板

在按住Win键的情况下,按Q键,呼唤出搜索框。
在搜索框中输入control panel,如下图所示:

 
技术图片
image.png

 

2.进入NVIDIA控制面板

在上图的搜索框中,已经显示出NVIDIA控制面板,如果读者有显示,则可以忽略第一步,直接点击进入NVIDIA控制面板。
下面2张图演示如何在控制面板中找出NVIDIA控制面板。
首先选择查看方式小图标,如下图红色方框标注处所示:

 
技术图片
image.png

然后在小图标中找出NVIDIA控制面板,如下图红色方框标注处所示:
 
技术图片
image.png

 

3.查看系统信息

选择进入导航栏的帮助中的系统信息,如下图红色方框标注处所示:

 
技术图片
image.png

 

4.查看组件

点击组件按钮,如下图红色箭头标注处所示。
在红色方框和红色文字共同标注处则显示当前电脑的CUDA版本号,例如本文作者是CUDA9.2。

 
技术图片
image.png

 

5.总结

1.2018年9月25日是中秋节假后三天第1天上班,写点简单的新手指南恢复学习状态。
2.知道自己电脑的CUDA版本号,则可以 选择合适版本的CUDA Toolkit,例如上图本文作者的CUDA版本号为9.2,则我可以安装CUDA Toolkit 9.2、CUDA Toolkit 9.0、CUDA Toolkit 9.1、CUDA Toolkit 8.0等。
3.查看CUDA Toolkit历史版本官方链接:https://developer.nvidia.com/cuda-toolkit-archive



作者:潇洒坤
链接:https://www.jianshu.com/p/d3b9419a0f89
来源:简书
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。

win10/windows 安装Pytorch

标签:index   targe   技术   data-   style   upload   imageview   cap   uda   

原文地址:https://www.cnblogs.com/xzdwn/p/12793977.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!