安装pytorch-gpu的经验与教训
2023-03-20 15:31:02 时间
首先说明
本文并不是安装教程,网上有很多,这里只是自己遇到的一些问题
我是以前安装的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的信息全部删除,再按照前面的流程走一遍,相信你自己一定会出现
相关文章
- 金融服务领域的大数据:即时分析
- 影响大数据、机器学习和人工智能未来发展的8个因素
- 从0开始构建一个属于你自己的PHP框架
- 如何将Hadoop集成到工作流程中?这6个优秀实践必看
- SEO公司使用大数据优化其模型的5种方法
- 关于Web Workers你需要了解的七件事
- 深入理解HTTPS原理、过程与实践
- 增强分析:数据和分析的未来
- PHP协程实现过程详解
- AI专家:大数据知识图谱——实战经验总结
- 关于PHP的错误机制总结
- 利用数据分析量化协同过滤算法的两大常见难题
- 怎么做大数据工作流调度系统?大厂架构师一语点破!
- 2019大数据处理必备的十大工具,从Linux到架构师必修
- OpenCV中的KMeans算法介绍与应用
- 教大家如果搭建一套phpstorm+wamp+xdebug调试PHP的环境
- CentOS下三种PHP拓展安装方法
- Go语言HTTP Server源码分析
- Go语言HTTP Server源码分析
- 2017年4月编程语言排行榜:Hack首次进入前五十