pytorch(GPU版)安装

确认有无英伟达显卡,有才能安装GPU版的pytorch,否则只能装CPU版
1.任务管理器->性能:

设备管理器->显示适配器,也可以:

nvidia驱动安装地址(大部分电脑自带,不需要额外安装):
https://www.nvidia.cn/Download/index.aspx?lang=cn


2.nvidia控制面板->帮助->系统信息->组件
查看nvidia驱动器版本,图中为11.7,即要安装的cuda版本不高于11.7即可

cmd/anaconda prompt也可:
nvidia-smi

这里nvcc -V一定是你电脑目前版本,nvidia-smi不一定
3.在官网找到自己对应的历史版本
https://developer.nvidia.com/cuda-toolkit-archive

选择一个版本下载

根据B站小土堆佬说,pytorch似乎已经集成了cudnn,不需要另外下载。
4.进入pytorch官网https://pytorch.org/

这里版本都高于11.7,选择历史版本

5.在命令行安装
以下提供镜像源:
清华镜像源:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
北京镜像源:https://mirrors.bfsu.edu.cn/anaconda/cloud/pytorch/win-64/
阿里镜像源:http://mirrors. aliyun.com/anaconda/cloud pytorch/win-64/
安装pytorch其实主要是安装torch、torchvision、torchaudio这三个包,torch最大一个多G,直接安装可能速度较慢

这一步另一个方法可以通过网址:https://download.pytorch.org/whl/torch/查找

ctrl+F搜索cu117,cu表示cuda,cp表示python,找到对应的链接,下载
下载完成后在命令行输入pip install 路径\文件名安装
pip install E:\Anaconda\torch-1.13.1+cu117-cp37-cp37m-win_amd64.whl
6.查看
如果conda list中pytorch后面版本号带有cpu,那就是中间哪儿出错了,装的cpu版本,卸载重来。
命令行python->import torch->torch.cuda.is_available()
返回True即安装成功

over ^o^/

posted @ 2024-05-20 19:33  Elite_youth  阅读(134)  评论(0编辑  收藏  举报