jax安装Ubuntu，cudnn版本查看

最新推荐文章于 2024-07-31 05:50:16 发布
最新推荐文章于 2024-07-31 05:50:16 发布
阅读量917 收藏
点赞数 1
分类专栏： jax 人工智能 cudnn 文章标签： ubuntu linux 运维
版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。
本文链接：https://blog.csdn.net/jueduke3483/article/details/129257460
版权
jax 同时被 3 个专栏收录
1 篇文章 0 订阅
订阅专栏
人工智能
1 篇文章 0 订阅
订阅专栏
cudnn
1 篇文章 0 订阅
订阅专栏

 

Ubuntu18.04，有GPU，jax安装后显示错误如图所示，无法识别GPU，使用升级后并不能解决

Cuda >= 11.8 and cudnn >= 8.6，采用如下方法重新安装

pip install "jax[cuda11_cudnn86]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
如果Cuda >= 11.4 and cudnn >= 8.2

pip install "jax[cuda11_cudnn82]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html

查看当前系统cudnn版本方法：

 vim /usr/local/cuda-11.6/include/cudnn_version.h

当前cudnn版本为8.4

 

 

 

                

        

        

    

  


    

      

        

            

              
                
                  空中守护者
                
              
            

            

                关注
              
            

        

        

          
      

      

            

                专栏目录
          









                



	

		

			

				
					
第4章: AI开发环境搭建

				
			

			

				

					AI天才研究院
				

				

					09-28
					
					1041
					
				

			

		

		

			
				
自然语言处理（NLP）是AI的重要应用领域，选择合适的NLP工具包可以大大提高开发效率。特点：全面的NLP功能丰富的语料库和词典适合教学和研究适用场景：文本分类、情感分析、词性标注spaCy特点：高性能，适合大规模处理提供预训练模型良好的深度学习集成适用场景：命名实体识别、依存句法分析、文本分类Gensim特点：专注于主题建模和文档相似性高效处理大规模文本数据支持词嵌入模型（Word2Vec, FastText等）

			
		

	



                

            
              



	

		

			

				
					
JAX 中文文档（一）

				
			

			

				

					龙哥盟
				

				

					06-21
					
					1386
					
				

			

		

		

			
				
原文：jax.readthedocs.io/en/latest/

开始入门
安装 JAX

原文：jax.readthedocs.io/en/latest/installation.html

使用 JAX 需要安装两个包：jax 是纯 Python 的跨平台库，jaxlib 包含编译的二进制文件，对于不同的操作系统和加速器需要不同的构建。
TL;DR 对于大多数用户来说，典型的 JAX 安装可能如下所示：


仅限 CPU（Linux/macOS/Windows）
pip install -U jax

			
		

	



              


  
              

                


	

		

			

				
					
jax 和 jaxlib 的 cuda 版本安装

				
			

			

				

					weixin_51793361的博客
				

				

					07-09
					
					2259
					
				

			

		

		

			
				
笔者花费时间才在 Ubuntu 20.04 适配上 jax 和 jaxlib 的 cuda 版本安装，以及 chex 版本。

			
		

	





	

		

			

				
					
ubuntu安装JAX GPU版本

				
			

			

				

					L0_L0的博客
				

				

					09-27
					
					3264
					
				

			

		

		

			
				
首先cuda是可以向下兼容的，因此，在ubuntu22.0.4上我的cuda为12.0，cudnn为8.8；注意自己安装的python版本，创建环境指令后面的版本号要与以安装的python版本一致。已经安装好cuda和对应的cudnn，以及anaconda。，自己查询符合自己安装条件的版本。环境中是否能够得到如下类似结果。测试如下文件在所创建的。

			
		

	



		

			
		



	

		

			

				
					
Ubuntu 20.04系统下安装GPU版本jax

				
			

			

				

					qq_52505378的博客
				

				

					06-23
					
					204
					
				

			

		

		

			
				
在此之前确定好自己电脑的CUDA版本以及CUDNN版本。按照我自己电脑的版本配置，我先后输入。中途没有报错就安装成功啦！

			
		

	





	

		

			

				
					
安装旧版本 jax 0.2.11 jaxlib 0.1.72 参考教程

				
			

			

				

					root0007的博客
				

				

					06-01
					
					856
					
				

			

		

		

			
				
要复现的工作需要用jax>=0.2.8版本，实际安装中遇到一些困难。现在将具体的版本记录在这里供大家参考。

			
		

	





	

		

			

				
					
ubuntu 安装 jax jaxlib cpu 和 gpu 版本 以及 tensorflow tensorRT的安装

				
			

			

				

					eloudy的专栏
				

				

					02-28
					
					1万+
					
				

			

		

		

			
				
安装jax jaxlib gpu  tensorflow gpu  tensorRT

			
		

	





	

		

			

				
					
linux下安装带cuda版本的jax

				
			

			

				

					07-28
				

			

		

		

			
				
- *3* [jax安装Ubuntu，cudnn版本查看](https://blog.csdn.net/jueduke3483/article/details/129257460)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_...

			
		

	





	

		

			

				
					
使用Flax、Jax、TF经历记录

				
			

			

				

					shenyh15的博客
				

				

					10-20
					
					2636
					
				

			

		

		

			
				
近期复现别人代码时需要使用flax、jax、tensorflow，在安装过程中遇到了一些bug，记录在此。

			
		

	





	

		

			

				
					
2080Ti深度学习环境配置及常用软件安装

				
			

			

				

					qq_30899353的博客
				

				

					08-28
					
					644
					
				

			

		

		

			
				
2080Ti深度学习环境配置及常用软件安装  ubuntu 16.04双系统的安装(已有window10系统)显卡驱动安装tensorflow安装teamviewer安装pycharm安装matlab安装

双系统的安装(已有window10系统)

ubuntu参考安装教程 [1]  [2] [3] [4]
ubunu16.04下载地址
U盘启动制作软件rufus

首先，准备软件和工具，下载ubuntu16.04 iso镜像文件，U盘启动软件rufus以及大于等于4G U盘一个（制盘时会格式化）
在自己

			
		

	





	

		

			

				
					
extending-jax:使用自定义C ++和CUDA代码扩展JAX

				
			

			

				

					04-22
				

			

		

		

			
				
使用自定义C ++和CUDA代码扩展JAX
 该存储库旨在作为一个教程，演示当您具有C ++和CUDA（可选）的现有实现时，提供JAX中的自定义操作所需的基础结构。 我本来想将其写为博客文章，但最终有足够的样板代码来决定，将其作为自述文件与自述文件中的回购库共享是更有意义的，所以我们到了！
 这样做的动机是，在我的工作中，我想使用像JAX这样的库来使模型适合天体物理学中的数据。 在这些模型中，通常至少有一部分模型规范是物理驱动的，尽管通常存在这些模型元素的实现，但是将它们重新实现为高级JAX函数通常效率低下或不切实际。 相反，我想将C中经过良好测试和优化的实现直接向JAX公开。 在我的工作中，这经常包括诸如迭代算法或特殊功能之类的东西，这些东西不太适合直接使用JAX来实现。
 因此，作为更新我的系库以与JAX交互的一部分，我必须学习支持该用例所需的基础结构，并且由于找不到一个可以在一处涵盖

			
		

	





	

		

			

				
					
【人工智能】大模型的涌现能力：概念、原理、本质和未来发展趋势

					
最新发布

				
			

			

				

					AI天才研究院
				

				

					07-31
					
					175
					
				

			

		

		

			
				
2023 年3 月的时候，由图灵奖得主Yoshua Bengio、特斯拉CEO马斯克、纽约大学名誉教授Gary Marcus、UC伯克利教授Stuart Russell等1000多人，在一封叫停GPT-4后续AI大模型的公开信上签名。这封公开信指出，最近人工智能陷入了一场失控的竞赛，模型的创造者们没有办法理解、预测或可靠地控制自己创造的大模型，人类社会对其可能造成的影响也没有做好准备。因此，公开信呼吁，所有AI实验室应立即暂停训练比GPT-4更强大的AI模型，为期至少6个月。

			
		

	





	

		

			

				
					
JAX: 库安装和GPU使用，解决不能识别gpu问题

				
			

			

				

					Papageno的博客
				

				

					07-13
					
					2万+
					
				

			

		

		

			
				
jax库安装后，不能识别gpu问题；cuda，cudnn版本问题。

			
		

	





	

		

			

				
					
JAX安装过程中遇到的各种坑

				
			

			

				

					xiaomianyuan的博客
				

				

					12-14
					
					6061
					
				

			

		

		

			
				
安装JAX时遇到的各种坑的解决办法

			
		

	





	

		

			

				
					
JAX安装

				
			

			

				

					weixin_50008473的博客
				

				

					08-30
					
					6029
					
				

			

		

		

			
				
本文安装版本为：jax0.2.21、jaxlib0.1.72、tensorflow2.7。

			
		

	





	

		

			

				
					
查看搭建的cuda环境

				
			

			

				

					m0_57782802的博客
				

				

					08-23
					
					788
					
				

			

		

		

			
				
cmd打开命令窗口，输入conda-env list 查看有几个conda环境。选择一个环境进入conda，输入conda activate pytorch。在Python中，查看torch、cuda和cudnn的版本。进入python环境，输入python。1. 查看torch版本。3.查看cudnn版本。2.查看cuda版本。

			
		

	





	

		

			

				
					
【Jax报错】CUDA backend failed to initialize: Unable to load cuPTI

				
			

			

				

					A91A981E的博客
				

				

					03-21
					
					884
					
				

			

		

		

			
				
如果没有显示，那要看你是不是安装的jax的gpu版本，参考。如题，在本地跑xai的grok-1的时候遇见的问题。首先你的cuda应该是安装好的，也就是bash下。应该是包含so文件的，这时候在。这样的，有几个卡就显示几个。可以显示、python下。

			
		

	





	

		

			

				
					
查看cuDNN的3种方法

					
热门推荐

				
			

			

				

					幸运的小菜鸟的博客
				

				

					09-11
					
					2万+
					
				

			

		

		

			
				
打开命令提示符或 PowerShell。这将显示 cuDNN 的版本号信息。

			
		

	





	

		

			

				
					
flax与jax、jaxlib的安装顺序和版本匹配

				
			

			

				

					syfei0518的博客
				

				

					06-16
					
					7188
					
				

			

		

		

			
				
在复现代码时需要用到这三个库，在安装的时候遇到了一些问题，现将解决措施记录如下。本文介绍了flax与jax、jaxlib的安装顺序和版本匹配，希望对大家有帮助。

			
		

	





	

		

			

				
					
ubuntu安装jax

				
			

			

				

					08-12
				

			

		

		

			
				
您可以在Ubuntu上安装JAX，具体步骤如下：

1. 首先，确保您的操作系统已经安装了Python和pip。

2. 如果您想安装支持CUDA 12的JAX，您可以运行以下命令：
   ```
   pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
   ```
   这将安装JAX和与CUDA 12兼容的扩展。

3. 如果您想安装支持CUDA 11的JAX，您可以运行以下命令：
   ```
   pip install --upgrade "jax[cuda11_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html
   ```
   这将安装JAX和与CUDA 11兼容的扩展。

4. 如果您只需要安装CPU版本的JAX，您可以运行以下命令：
   ```
   pip install --upgrade pip
   pip install --upgrade "jax<span class="em">1</span><span class="em">2</span><span class="em">3</span>
#### 引用[.reference_title]
-  *1* *2* [Ubuntu22.04安装Whisper-jax](https://blog.csdn.net/qq_43907505/article/details/130866691)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
-  *3* [Ubuntu下jax安装与使用](https://blog.csdn.net/zaf0516/article/details/126390534)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v92^chatsearchT3_1"}}] [.reference_item style="max-width: 50%"]
[ .reference_list ]

			
		

	



              




          




        



    





    



      

      

        
      

      





	

		评论	
	

	
	




  

    

      添加红包
      
    

    

      

        
        

          
          
        

        
请填写红包祝福语或标题

      

      

        
        

          
          个
        

        
红包个数最小为10个

      

      

        
        

          
          元
        

        
红包金额最低5元

      

      

        
        

          当前余额3.43元
          前往充值 >
        

      

      

        

          需支付：10.00元
        

        
        
      

    

  





  

    
    
  

  

    
    
  








  

    

      

        

          

            
            
成就一亿技术人!

          

          

        

        

          

          

            领取后你会自动成为博主和红包主的粉丝
            规则
          

        

      

      

        

          

            
              
            
            

              
hope_wisdom
 发出的红包
            

          

        

        

          

          

          

        

      

    

    

  



      
      

      
实付元

      
使用余额支付

      

        

          

            
            点击重新获取
          

        

        
扫码支付

      

      

        
          
            
          
          
        
      

      

        
        钱包余额
          0
          

            
            

              

                
抵扣说明：

                
 1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
 2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。

              

            

          

      

      余额充值