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^/