当前位置:   article > 正文

pytorch和pytorchvision安装

pytorch和pytorchvision安装

参考https://blog.csdn.net/2301_76863102/article/details/129369549
https://blog.csdn.net/weixin_43798572/article/details/123122477
查看我的版本
右键,nvivdia控制面板,帮助,系统信息
在这里插入图片描述
驱动程序版本号为528.49 更新很快的
在这里插入图片描述

CUDA版本12.0.147
代表我的CUDA Toolkit是12.0,我不用下载(直接安装pytorch,下面有),
但是,如果复现论文,则可以按要求来
我们开始下载CUDA Toolkit
CUDAToolkit下载地址https://developer.nvidia.com/cuda-toolkit-archive
首先查看文档https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
在这里插入图片描述我应该下载12.1
但是目前pytorch最高是11.8,我们要和它匹配,所以也下载11.8
在这里插入图片描述
在这里插入图片描述

安装pytorch

在这里插入图片描述在这里插入图片描述报错,直接删掉这两个文件
在这里插入图片描述本次我直接安装pytorch,没诶

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/空白诗007/article/detail/975299
推荐阅读
相关标签
  

闽ICP备14008679号