首页 > 其他分享 >win11_安装VScommunity|CUDA|pytorch

win11_安装VScommunity|CUDA|pytorch

时间:2023-02-24 17:44:26浏览次数:38  
标签:Visual pytorch Studio CUDA win11 安装 cuda

注意:安装cuda过程中,有一步会检查是否安装了Visual Studio,所以本文中先安装Visual Studio,再安装cuda。

参考:https://blog.csdn.net/weixin_44019018/article/details/122052736

一、Visual Studio安装

下载地址:https://learn.microsoft.com/zh-cn/visualstudio/releases/2019/release-notes#16.11.24

 

 1 双击引导程序 VisualStudioSetup.exe 开始安装

 

2 安装Visual Studio 安装程序后,可以通过选择所需的功能集或工作负载来使用它来自定义安装。 操作方法如下。

 在 Visual Studio 安装程序中选择所需的工作负载。(我选了支持python的工作负荷,以及面向C++项目开发跨平台应用)

 

 二、CUDA Toolkit 安装

1 查看自己电脑的显卡驱动

 

2 下载驱动

需要下载官方推荐的适合的驱动。地址:https://www.nvidia.cn/Download/index.aspx?lang=cn#

注意:下载时需要有NVIDIA官网的帐号,登录之后才能下载

 

 

 

3 显卡驱动支持的cuda版本查看

 

 4 cuda 版本查看

根据我要装的pytorch支持的cuda版本、并且cuda版本要在我自己电脑的驱动版本之下,选择安装cuda11.7

 pytorch对应的cuda版本查看:https://pytorch.org/get-started/locally/

 

 

 

 5. cuda toolkit 下载

地址:https://developer.nvidia.com/cuda-toolkit-archive

选择相应版本号

 

 

 

 

 

 

 

6. 安装 cuda toolkit

 双击运行程序,开始安装

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 

 用命令行检查下:nvcc --version

 

 

 

 7 cudnn安装

将下载的安装包解压

将文件中的三个子文件夹复制到路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7

 

 

 

 添加环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.7\lib

 

 

 三、pytorch安装

 

 按照网站中给出的指令进行安装,安装过程较慢,请耐心等待,安装完成后按照下述方式进行测试。

 

 

标签:Visual,pytorch,Studio,CUDA,win11,安装,cuda
From: https://www.cnblogs.com/lynsha/p/17152584.html

相关文章

  • 【pytorch】pytorch自定义训练vgg16和测试数据集 微调resnet18全连接层
    自己定义模型测试:correct=0total=0fordataintest_loader:img,label=dataoutputs=net(Variable(img))_,predict=......
  • 基于pytorch实现模型剪枝
    一,剪枝分类1.1,非结构化剪枝1.2,结构化剪枝1.3,本地与全局修剪二,PyTorch的剪枝2.1,pytorch剪枝工作原理2.2,局部剪枝2.2.1,局部非结构化剪枝2.2.2,局部结构化剪枝......
  • Python中用PyTorch机器学习神经网络分类预测银行客户流失模型|附代码数据
    阅读全文:http://tecdat.cn/?p=8522最近我们被客户要求撰写关于神经网络的研究报告,包括一些图形和统计输出。分类问题属于机器学习问题的类别,其中给定一组特征,任务是预测......
  • CPU推理|使用英特尔 Sapphire Rapids 加速 PyTorch Transformers
    在最近的一篇文章中,我们介绍了代号为SapphireRapids的第四代英特尔至强CPU及其新的先进矩阵扩展(AMX)指令集。通过使用AmazonEC2上的SapphireRapids服务器......
  • pytorch----.cuda()的作用
    使用Pytorch框架时,会看到某个变量后面加上如a.cuda(),那么cuda()的作用是什么呢?一、cuda()函数的作用在pytorch中,即使是有GPU的机器,它也不会自动使用GPU,而是需要在程序中显示......
  • pytorch反向接口暴露
    主目录下CMakeList.txt嵌套Caffe目录Caffe目录包含aten,***等目录Caffe同时调用Codegen.cmake执行动态生成代码,生成动态变量所以aten目录下可以调用动态变量并构建,......
  • Linux+conda+虚拟环境中安装cuda及cudnn
    经过一番折腾得出几点总结:conda管理下的虚拟环境中单独安装cudnn无法做到普通环境中的安装:解压而后放到对应的cuda文件夹中,因为虚拟环境安装下cuda和普通环境中安装的目......
  • PyTorch使用预训练模型
    ​​​​​​state_dict()功能:获取模型当前的参数,以一个有序字典形式返回。这个有序字典中,key是各层参数名,value就是参数。load_state_dict(state_dict)功能:将state......
  • PyTorch——SummaryWriter,TensorboardX
    Tensorboard的工作流程将代码运行过程中的,某些你关心的数据保存在一个文件夹中:这一步由代码中的writer完成再读取这个文件夹中的数据,用浏览器显示出来:这一步通过在命令行运......
  • cuda、cudnn、tnesorrt的查看安装
    1、首先本地查看cuda已安装的版本11.7输入命令:【nvcc-V】输出:nvcc:NVIDIA(R)CudacompilerdriverCopyright(c)2005-2022NVIDIACorporationBuiltonWed_Jun__8_......