标签:torch cu113 pytorch 如何 PyTorch CUDA 版本 安装
前言
现在使用 pytorch 框架进行深度学习的人越来越多了,但是 torch 的安装(GPU 版本)和普通第三方包的安装过程不太一样,因为涉及到电脑的显卡驱动版本,下面来看看安装 pytorch 的正确姿势。
安装过程
查看驱动版本
打开 NVIDIA 控制面板,点击工具栏帮助 --> 系统信息,在弹出的对话框中有如下信息:
可以看到本机驱动程序版本为 466.92,接着前往 NVIDIA CUDA Toolkit Release Notes,在 cuda driver
这一节有下面这个表:
由于本机为 Windows 系统且驱动版本为 466.92,所以应该选择 CUDA 11.6.x。
下载 torch
在 PyTorch 官网的 Get Started 页面有下图所示的安装向导:
虽然没找到 CUDA 11.6.x,但是 CUDA 11.3 也是可以的,复制 Run this Command
里面的指令:
pip install torch==1.10.2+cu113 torchvision==0.11.3+cu113 torchaudio===0.10.2+cu113 -f https://download.pytorch.org/whl/cu113/torch_stable.html
这里把 pip3 换成了 pip,不过 pip3 也是可以的。在终端运行上述指令,等待安装完成即可。
标签:torch,cu113,pytorch,如何,PyTorch,CUDA,版本,安装 来源: https://www.cnblogs.com/zhiyiYo/p/15865454.html
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。