pip 安装GPU版本pytorch 与cuda下载

pytorch安装

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

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

pip 安装GPU版本pytorch 与cuda下载_第1张图片
这个意思是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的
pip 安装GPU版本pytorch 与cuda下载_第2张图片

pip 安装GPU版本pytorch 与cuda下载_第3张图片
然后我就换成10.2的试试,一路next安装,下载cudnn包,解压,复制到cuda目录下,还是不可用。最后换成GPU版的pytorch就解决了。
好了,回到这里,下一步怎么办,点击要下载的蓝字,弹出下载任务框
pip 安装GPU版本pytorch 与cuda下载_第4张图片
复制网址,放到下面模板中:

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就行了
在这里插入图片描述
最后等它安装完就好了。

cuda安装

网上搜索cuda,然后下载就行了
pip 安装GPU版本pytorch 与cuda下载_第5张图片
安装的话双击,然后一路next就行了
cudnn在
pip 安装GPU版本pytorch 与cuda下载_第6张图片
pip 安装GPU版本pytorch 与cuda下载_第7张图片
pip 安装GPU版本pytorch 与cuda下载_第8张图片
注册完下载与cuda对应的版本,解压。
复制下面三个文件
pip 安装GPU版本pytorch 与cuda下载_第9张图片
找到下面这个目录
pip 安装GPU版本pytorch 与cuda下载_第10张图片
复制进去(合并相同的文件夹)就行了
最后验证一下cuda是否可用
pip 安装GPU版本pytorch 与cuda下载_第11张图片

你可能感兴趣的:(安装)