pytorch-深度学习环境配置遇到的问题

最新推荐文章于 2024-07-01 21:06:51 发布

HFHHVINCENT

最新推荐文章于 2024-07-01 21:06:51 发布

阅读量836

点赞数

文章标签： pytorch

本文链接：https://blog.csdn.net/HFHHVINCENT/article/details/122055536

版权

本文介绍了在配置PyTorch深度学习环境时遇到的问题，包括Anaconda3、CUDA10.2、cuDNN和PyCharm的安装。特别提醒，从清华源镜像下载GPU版本PyTorch可能得到CPU版本。另外，环境激活和版本选择（如torch1.8.0+torchvision0.9.0与torch1.9.0+torchvision0.10.0）也会影响结果。在训练过程中遇到的错误解决办法是添加相关环境变量。

摘要由CSDN通过智能技术生成

版本 anaconda3.0 cuda10.2 cudnn python=3.9 torch1.8.0+torchvision0.9.0

anaconda3 +cuda10.2+ cudnn+pycharm 官网下载，比较容易
选择对应cuda10.2版本的torch1.8.0+torchvision0.9.0 pytorch轮子文件
清华源镜像gpu版本会出现下载的是cpu版本（花了我两天时间，看了一个老哥的博客我才知道）
torch1.9.0+torchvision0.10.0可能会出现（果然是这版本出现的问题）

RuntimeError: Couldn't load custom C++ ops. This can happen if your PyTorch and torchvision versions are incompatible, or

最低0.47元/天解锁文章

HFHHVINCENT

关注

0
点赞
踩
0

收藏

觉得还不错? 一键收藏
0
评论
pytorch-深度学习环境配置遇到的问题

版本 anaconda3.0 cuda10.2 cudnn python=3.9 torch1.8.0+torchvision0.9.0anaconda3 +cuda10.2+ cudnn+pycharm 官网下载，比较容易选择对应cuda10.2版本的torch1.8.0+torchvision0.9.0 pytorch轮子文件清华源镜像gpu版本会出现下载的是cpu版本（花了我两天时间，看了一个老哥的博客我才知道）torch1.9.0+torchvision0.10.0可能会出
复制链接

扫一扫