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安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
最近公司数据部那边有个需求，需要结合矢量数据和影像数据，进行批量截图，并且截图中只能有一个图斑，还要添加上相应的水印，这篇文章主要介绍了arcgis使用Python脚本进行批量截图,需要的朋友可以参考下2023-01-01
这篇文章主要介绍了python 定时器每天就执行一次的实现代码，代码简单易懂非常不错，具有一定的参考借鉴价值,需要的朋友可以参考下2019-08-08
这篇文章主要介绍了简单了解python反射机制的一些知识，反射机制就是在运行时，动态的确定对象的类型，并可以通过字符串调用对象属性、方法、导入模块，是一种基于字符串的事件驱动。,需要的朋友可以参考下2019-07-07
这篇文章主要介绍了python中的list字符串元素排序方式，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2022-11-11
本篇文章主要介绍了python并发2之使用asyncio处理并发，小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2017-12-12
在日常编程中,我们经常会遇到文件处理的需求,如base64与图片的相互转换、图片与像素点的操作、本地文件与二进制的互相转换、计算文件的md5以及下载网络文件等,这些操作对于处理多媒体数据、实现数据的持久化存储、保证数据的完整性和安全性等方面都至关重要2024-09-09
这篇文章主要介绍了Python编程使用NLTK进行自然语言处理详解，涉及了nltk和开发环境的简单介绍，以及SentencesSegment，SentencesSegment等内容，具有一定参考价值，需要的朋友可以了解下。2017-11-11
Diango + uwsgi + nginx项目部署的全过程（可外网访问）
这篇文章主要给大家介绍了关于Diango + uwsgi + nginx项目部署的全过程（可外网访问），文中通过示例代码将部署的过程介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧。2018-04-04
这篇文章主要介绍了Python编程pygal绘图实例之XY线，具有一定借鉴价值，需要的朋友可以参考下。2017-12-12
这篇文章主要介绍了Django 连接sql server数据库的方法,小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2018-06-06
最新评论