- 博客(3)
- 收藏
- 关注
RSS订阅原创 解决NVIDIA Geforce RIX 3060 Laptop GPU与pytorch版本不兼容
造成这一错误的原因是我的RTX3060的显卡和pytorch的版本不适配，pytorch的版本过低或者过高。可知当前python解释器的版本为3.8，torch的版本为1.10.0，cuda的版本为10.2，版本过低。其中torch的版本为1.13.0，适配于python3.8的版本。找到对应的CUDA版本，我对应下载CUDA11.6版本，需要注意的是版本之间的对应，其官网的对应关系如下链接。卸载完成之后需要重新安装适配显卡的torch版本。显卡所支持cuda的最高版本为11.7。安装成功，问题解决。
2024-03-07 22:22:43
820
原创 next(iter(dataloader))的详解
经常在自己建立好数据集后需要验证一下，自己的图片和标签能不能对上，经常用到next(iter(dataloader))命令。可知其返的是一个batch的图片和标签数据，数据类型均为tensor，图片维度为（c，h，w）。分别打印出返回值的大小、类别。
2023-09-27 14:42:01
767
1
原创 使用pycharm配置深度学习环境
在下载时候一定采用pip下载，不要用conda，否则下载的是cpu版，会造成torch.cuda.is_available() 的输出无论如何都是False，其中+cu102表示cuda版本为10.2，一定要带上。在安装torch时候，需要注意的是版本之间的对应，需要版本一致，其官网的对应关系如下链接。由于我创建虚拟环境是python3.8，因此我选用10.2的版本的cuda进行下载，创建虚拟环境，我创建的是python3.8版本的，版本不宜过高。一、查看可支持cuda版本。输出True，安装成功。
2023-09-26 13:56:12
595
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人