安装pytorch-gpu的经验与教训
2024-09-05 19:21:56
首先说明
本文并不是安装教程,网上有很多,这里只是自己遇到的一些问题
我是以前安装的tensorflow-gpu的,但是发现现在的学术论文大部分都是用pytorch复现的,因此才去安装的pytorch-gpu
查看自己安装的CUDA
nvcc -V
我的是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的信息全部删除,再按照前面的流程走一遍,相信你自己一定会出现
最新文章
- 15、java中的内部类介绍
- nil与NULL的区别
- JAVA List<;>; 合并去重
- VIM常用快捷键~网页上查找
- Hadoop MRUnit使用(一)
- 分布式MySQL集群方案的探索与思考
- 自定义TexturePacker插件导出自己的plist文件
- HttpRequest Get Post,WebClient Get
- php后台如何避免用户直接进入方法实例
- HTML5之 WebSockets
- [转]Mac App distribution in App Store
- [置顶] android AIDL 进程间通信
- sharePoint常用命令
- TextView赋值int型,并显示
- iptables防火墙详解
- 《Unity3D-播放被打中的时候粒子的特效的代码》
- Jenkins结合.net平台之Web项目编译
- Iview Modal 点击确定就关闭,我想禁止关闭该怎么办呢?
- Beta冲刺阶段3.0
- 解题:POJ 2888 Magic Bracelet
热门文章
- 齐博X1模板页面之间的继承关系
- Istio(二):在Kubernetes(k8s)集群上安装部署istio1.14
- 十二、Pod的NameSpace
- Java线程未捕获异常处理 UncaughtExceptionHandler
- 「浙江理工大学ACM入队200题系列」问题 H: 零基础学C/C++18——三位数反转
- go GMP
- kubeedge的云边协同通道
- Java—猜数字
- element-ui 对话框dialog里使用echarts,报错&#39;dom没有获取到&#39;?
- dp状态设计