1.确认显卡是否NVIDIA,是的话才能装gpu版的,怎么判断电脑显卡是NVIDIA?
对我的电脑点击右键——管理——设备管理器——显示适配器,可以显示显卡的型号
点击显示适配器前面的>
这样就能看到你的显卡是什么型号
然后关键一步来了,到官网下载驱动
https://www.nvidia.cn/Download/index.aspx?lang=cn
按照的产品类型型号选择下载即可
2.驱动装好后,打开Anaconda3
执行语句创建环境:conda create -n pytorch python=3.7
然后登录:https://pytorch.org/get-started/locally/
向下滑动能看到
按照你的配置选择,我的选择如下:
复制上面箭头指的语句到刚才创建环境的位置Anaconda Powershell Prompt执行箭头指的语句conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
等待下载即可,速度很慢,可以选择早上下载,如果不行可以手机开热点给电脑,总共需要2G多的流量
3.装好后怎么判断是否装好pytorch-gpu?
>>import torch
>>torch.cuda.is_available()
True
这样就安装好了