首页 > 其他分享 >升级显卡驱动,下载cuda和cudann 12.0后,安装pytorch

升级显卡驱动,下载cuda和cudann 12.0后,安装pytorch

时间:2023-11-05 18:45:11浏览次数:38  
标签:知乎 PyTorch pytorch 12.0 cuda 安装

1、重新创建虚拟环境

 2、查找pytorch版本

安装Pytorch如何选择CUDA的版本,看这一篇就够了 - 知乎 (zhihu.com)

Previous PyTorch Versions | PyTorch

安装支持CUDA 12的pytorch教程 - 知乎 (zhihu.com)

 

于是:

 

 conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia

 (未开net,因为昨天开着报错了,真的好慢,还怕清华源变成CPU的...)

 完成

试了一下啊,cuda可用。

标签:知乎,PyTorch,pytorch,12.0,cuda,安装
From: https://www.cnblogs.com/hxpp/p/17810830.html

相关文章

  • Pytorch 模型文件后缀名含义
    目前常见的几种pytorch模型后缀名有:.pt.pth.bin.onnx 其实,.pt.pth.bin这三个后缀都是人为指定的,在保存模型的时候没有区别,但用不同后缀是为了方便区分它们所储存的内容,相当于是个标记:格式解释适用场景可对应的后缀.pt或.pthPyTorch的默认模型文件......
  • cuda安装记录
    1、显卡驱动程序更新      2、安装cuda https://developer.nvidia.com/cuda-toolkit-archive          3、安装cudann重启电脑cuDNNArchive|NVIDIADeveloper    ......
  • 深度解析BERT:从理论到Pytorch实战
    本文从BERT的基本概念和架构开始,详细讲解了其预训练和微调机制,并通过Python和PyTorch代码示例展示了如何在实际应用中使用这一模型。我们探讨了BERT的核心特点,包括其强大的注意力机制和与其他Transformer架构的差异。关注TechLead,分享AI全维度知识。作者拥有10+年互联网服务架......
  • pytorch图像变换和增强
    pytorch图像变换和增强目录pytorch图像变换和增强总览调整大小灰度变换标准化水平垂直翻转随机旋转中心裁剪随机裁剪亮度对比度饱和度高斯模糊高斯噪声随机块中心区域参考资料总览#使用数据增强技术可以增加数据集中图像的多样性,从而提高模型的性能和泛化能力。1.尺寸变换tr......
  • 《深度学习的数学》(涌井良幸、涌井贞美著) 神经网络计算pytorch示例二
    涌井良幸、涌井贞美著的《深度学习的数学》这本书,浅显易懂。书中还用Excel示例(如下图)神经网络的计算,真是不错。但光有Excel示例还是有点欠缺的,如果有pytorch代码演示就更好了。百度了半天在网上没找到别人写的,只好自己撸一个(使用python+pytorch),供同样在学习神经网络的初学者参考......
  • 深度学习中基于pytorch的多层感知机简洁实现
    基于一个例子,总结过去一个月的学习:importtorchfromtorchimportnnfromtorch.nnimportinitimportnumpyasnpimportsyssys.path.append("..")importd2lzh_pytorchasd2lnum_inputs,num_outputs,num_hiddens=784,10,256net=nn.Sequential(d2......
  • TensorFlow、PyTorch、Keras、Scikit-learn和ChatGPT。视觉开发软件工具 Halcon、Visi
     目录TensorFlow、PyTorch、Keras、Scikit-learn和ChatGPT1.TensorFlow2.PyTorch3.Keras视觉开发软件工具Halcon、VisionPro、LabView、OpenCV,还有eVision、Mil、Sapera等。(一)、Halcon(二)OpenCV:ComputerVision(计算机视觉)(三)VisionProTensorFlow、PyTorch、Keras、Scikit-learn和......
  • 【深度学习】PyTorch的基本运算 与 构造简单神经网络模型
    基本运算importtorch#创建一个自定义的张量t=torch.tensor([1.0,2.0,3.0])#tensor([1.,2.,3.])#求平均值t.mean()#tensor(2.)#创建一个指定行列的张量x=torch.empty(3,5)#tensor([[0.,0.,0.,0.,0.],[0.,0.,0.,0.,0.],[0.,0.,0.,0.,0.]......
  • 【转载】CUDA编程学习记录 C++
    参考Yuezero的CUDA编程基础(https://blog.csdn.net/weixin_54338498/article/details/127947551)CUDA编程模型host指代CPU及其内存,包含host程序device指代GPU及其内存,包含device程序经典CUDA程序的执行流程如下:分配host内存,并进行数据初始化;分配device内存,并从host将......
  • 如何查看windows的CUDA版本
    打开CMD命令行窗口,输入nvcc--version nvcc--versionnvcc:NVIDIA(R)CudacompilerdriverCopyright(c)2005-2020NVIDIACorporationBuiltonWed_Jul_22_19:09:35_Pacific_Daylight_Time_2020Cudacompilationtools,release11.0,V11.0.221Buildcuda_11.0_bu.relgpu_......