gpt4 book ai didi

安装pytorch-gpu的经验与教训

转载 作者:我是一只小鸟 更新时间:2023-01-12 22:32:05 25 4
gpt4 key购买 nike

首先说明

本文并不是安装教程,网上有很多,这里只是自己遇到的一些问题 。

我是以前安装的tensorflow-gpu的,但是发现现在的学术论文大部分都是用pytorch复现的,因此才去安装的pytorch-gpu 。

查看自己安装的CUDA 。

                          nvcc -V
                        

这里我提供一个安装tensorflow时所用的CUDA对应表 。

 安装cuDNN时版本一定要对应正确 。

  。

 安装完CUDA时要改一下环境变量 。

  。

 只用添加这个就好,其他三个CUDA会自动为你添加 。

 我的是cuda-11.2 。

但是官网没有配套的,直接就挑自己喜欢的下就行 Start Locally | PyTorch 。

  。

 这里强调:千万不要相信在其他网站上的下载命令。有可能你下载的就是cpu版本

安装的时候可能会遇到 。

Solving environment: failed with initial frozen solve. Retrying with flexible solve.

不要担心,不用管它,耐心等一等,就会出现(y/n)选择y就可以了 。

这里要看一眼你的pytorch是不是cuda版本的,不是要重新下载

测试安装是否成功 。

                          
                            import
                          
                          
                             torch

                          
                          
                            print
                          
                          (torch.
                          
                            __version__
                          
                          )  
                          
                            #
                          
                          
                            注意是双下划线
                          
                          
                            print
                          
                          
                            (torch.version.cuda)

                          
                          
                            print
                          
                          
                            (torch.cuda.is_available())

                          
                          
                            print
                          
                          (torch.cuda.get_device_name())
                        

当然你这些都做完了,还是出现了False 。

失败后的操作

打开你的anaconda 。

  。

 将这里关于pytorch的信息全部删除,再按照前面的流程走一遍,相信你自己一定会出现 。

  。

最后此篇关于安装pytorch-gpu的经验与教训的文章就讲到这里了,如果你想了解更多关于安装pytorch-gpu的经验与教训的内容请搜索CFSDN的文章或继续浏览相关文章,希望大家以后支持我的博客! 。

25 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com