pip 安装GPU版本pytorch 与cuda下载

最新推荐文章于 2024-10-01 23:59:28 发布

谨慎的海绵

最新推荐文章于 2024-10-01 23:59:28 发布

阅读量5.3k

点赞数 4

分类专栏：安装

本文链接：https://blog.csdn.net/asapigi/article/details/119713047

版权

安装专栏收录该内容

5 篇文章 0 订阅

订阅专栏

pytorch安装

不能访问外网，连pytorch官网也打不开
访问这个网站

https://download.pytorch.org/whl/torch_stable.html

在这里插入图片描述
这个意思是cpu版本，对应pytorch版本为1.8.0，对应python版本为3.8，对应操作系统是Mac。
根据我自己的电脑，下载对应版本：

cu102指的是cuda版本为10.2，这里为什么我不下最新的11.4呢
因为我之下了11.4的然后torch一直说不可用，这里我不知道pytorch还有CPU和GPU版本的区别，然后控制面板告诉我，我的GPU是可以用11.4的cuda的
在这里插入图片描述

在这里插入图片描述
然后我就换成10.2的试试，一路next安装，下载cudnn包，解压，复制到cuda目录下，还是不可用。最后换成GPU版的pytorch就解决了。
好了，回到这里，下一步怎么办，点击要下载的蓝字，弹出下载任务框

复制网址，放到下面模板中：

pip install https://download.pytorch.org/whl/cu102/torch-1.9.0%2Bcu102-cp39-cp39-win_amd64.whl -i https://pypi.tuna.tsinghua.edu.cn/simple some-package

-i 前面的就是我们的下载地址，-i后面的是清华源镜像网站，这样就不用直接在外网上下了。
如果python版本不对应会报以下错误：
ERROR: torch-1.9.0+cu102-cp37-cp37m-win_amd64.whl is not a supported wheel on this platform.
查看python版本然后找到对应的pytorch就行了
在这里插入图片描述
最后等它安装完就好了。