2022-03-241 pytorch-gpu安装

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

这样就安装好了

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容