pytorch-gpu安装的经验与教训
更新时间：2023年01月13日 08:23:56 作者：故y
本文主要介绍了pytorch-gpu安装的经验与教训，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
首先说明
本文并不是安装教程，网上有很多，这里只是自己遇到的一些问题
我是以前安装的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安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了Python正则表达式实现截取成对括号的方法,涉及Python正则匹配相关操作技巧,需要的朋友可以参考下2017-01-01
这篇文章主要介绍了python 五子棋如何获得鼠标点击坐标,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-11-11
今天小编就为大家分享一篇OpenCV2从摄像头获取帧并写入视频文件的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-08-08
在日常工作中,我们经常需要处理和转换不同格式的Excel文件,以适应不同的需求和软件兼容性,Excel文件的两种常见格式是XLS（Excel 97-2003）和XLSX（Excel 2007及以上版本）,本文将详细介绍如何使用Python在XLS和XLSX格式之间进行转换,需要的朋友可以参考下2024-09-09
pytorch DataLoader的num_workers参数与设置大小详解
这篇文章主要介绍了pytorch DataLoader的num_workers参数与设置大小详解，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2021-05-05
今天给大家带来的是关于Python机器学习的相关知识,文章围绕着Python底层实现KNN展开,文中有非常详细的解释及代码示例,需要的朋友可以参考下2021-06-06
在Python中,我们可以通过__set_name__方法来实现一些特殊的操作,本文主要介绍如何在Python中实现__set_name__方法,并且给出一些实际应用的示例,感兴趣的可以了解一下2024-01-01
Windows下Python的Django框架环境部署及应用编写入门
这篇文章主要介绍了Windows下Python的Django框架环境部署及程序编写入门,Django在Python的框架中算是一个重量级的MVC框架,本文将从程序部署开始讲到hellow world web应用的编写,需要的朋友可以参考下2016-03-03
本文介绍了Python中列表的逆序、复制和清除操作,通过reverse()方法、切片、copy()方法和clear()方法,我们可以轻松地对列表进行这些操作2024-12-12
关于Python中的向量相加和numpy中的向量相加效率对比
今天小编就为大家分享一篇关于Python中的向量相加和numpy中的向量相加效率对比，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-08-08
最新评论