首页 > 其他分享 >conda安装gpu版本pytorch与gpu版本tensorflow

conda安装gpu版本pytorch与gpu版本tensorflow

时间:2024-02-14 23:22:10浏览次数:25  
标签:cudatoolkit cudnn pytorch 版本 tensorflow gpu 安装

创建环境进入环境
nvidia-smi查看cuda版本,根据cuda版本安装对应版本的pytorch,在pytorch官网可以查看,版本不合适可以使用较低版本cuda的torch,使用官网提供的命令行安装即可,import torch``print(torch.cuda.is_available())验证安装结果。
tensorflow的安装要在环境中安装cudatoolkit和cudnn,可以使用conda search 了解conda提供的cudatoolkit和cudnn版本,并做好版本对应,tensorflow/python/cudatoolkit/cudnn对应版本可以在官网查看https://tensorflow.google.cn/install/source_windows,安装顺序是cudatoolkit/cudnn/tensorflow,安装后import tensorflow as tf``print(tf.__version__)验证安装结果。

标签:cudatoolkit,cudnn,pytorch,版本,tensorflow,gpu,安装
From: https://www.cnblogs.com/ducn/p/18015846

相关文章

  • pytorch深度学习入门(8)之-Torchaudio使用Tacotron2 文本转语音
    https://blog.csdn.net/ajunbin859/article/details/134380417?ops_request_misc=&request_id=&biz_id=102&utm_term=pytorch%E7%89%88%E6%9C%AC%E7%9A%84tacotron%E8%AF%A6%E7%BB%86%E5%AE%89%E8%A3%85%E6%95%99%E7%A8%8B&utm_medium=distribute.pc_search_r......
  • Tomcat官网下载版本:如何选择tar.gz (pgp, sha512)和zip (pgp, sha512)?
    先上结论:选择tar.gz格式还是zip格式的文件下载,主要取决于操作系统和个人偏好:tar.gz(pgp,sha512):这是一种在Unix-like系统(如Linux和macOS)中常用的压缩格式。tar是一种将多个文件合并为单个文件(归档)的工具,而gz是gzip,用于压缩归档文件。pgp和sha512分别提供了文件......
  • pytorch3d 安装
    分享一下在Ubuntu20.04系统上安装PyTorch3D的经验,尤其是当遵循官方教程却遇到安装困难时。PyTorch3D是由FacebookAIResearch团队开发的,旨在支持与3D数据相关的深度学习任务,包括但不限于3D形状的建模、渲染和视觉理解。准备工作在安装PyTorch3D之前,首先需要确保您的系统已安装......
  • 17.Jenkins的Git版本管理
    Git的集成 在Jenkins上构建从Git源码仓库拉取代码的任务时需要使用到Git插件在执行job的机器上安装好Git yuminstallgit-y下载Git插件 进入菜单Dashboard->系统管理(ManageJenkins)->插件管理(ManagePlugins)在可选插件搜索Git找到Git......
  • springboot自定义starter 版本大于2.7
    相关jar包<!--提示配置--><dependency><groupId>org.springframework.boot</groupId><artifactId>spring-boot-configuration-processor</artifactId><version>3.2.1</versi......
  • 国产AI训练卡,对标美国NVIDIA公司的A100,华为昇腾Atlas 300T A2(Ascend 910B4)高性能GPU/N
    ChinahassuccessfullyachievedthelocalizationofAIchips,breakingthroughthetechnologicalrestrictionsimposedbytheU.S.governmentandrealizingindependentdesignandproductionofdomesticAIchips.Huawei'sAscend910modelAIchiphass......
  • SharePoint Online 对比文档库中文档的版本
    前言我们经常会使用SharePointOnline来保存和协作编辑文档,但是,有时候编辑的时候会想知道之前某个版本做过哪些更改,该如何去做呢?其实,Word本身为我们提供了一个非常好的比较功能,我们演示一下给大家看。正文1.需要使用Openinapp在本地word中打开文档,如下......
  • 165. 比较版本号(中)
    目录题目题解题目给你两个版本号version1和version2,请你比较它们。版本号由一个或多个修订号组成,各修订号由一个'.'连接。每个修订号由多位数字组成,可能包含前导零。每个版本号至少包含一个字符。修订号从左到右编号,下标从0开始,最左边的修订号下标为0,下一个修订......
  • IDEA下载其他版本及快速破解
    其他版本的下载IDEA下载-https://www.jetbrains.com/zh-cn/idea/download/other.htmljetbrains公司其他产品也是类似的路径,如Pycharm下载-https://www.jetbrains.com/zh-cn/pycharm/download/other.html,将路径中的idea换成pycharm。安装及破解安装就是正常安装,一直下一步就行,......
  • java中使用opencl操作GPU
    需要管理GPU资源,使用java编写,选用opencl框架,并且选择org.jocl包(<dependency><groupId>org.jocl</groupId><artifactId>jocl</artifactId><version>2.0.5</version></dependency>)。具体opencl原理此处不涉及,仅记录使用java该如何做基本操作。最少要以下几步,详细可以参看:ht......