当前位置：网站首页>深度学习------pytorch-gpu环境搭建

深度学习------pytorch-gpu环境搭建

2022-08-04 11:21:00 【小飞龙程序员】

本人笔记本1650显卡，pytorch（gpu）环境搭建
在这里插入图片描述

cuda10.2gpu安装命令：pip install torch1.10.1+cu102 torchvision0.11.2+cu102 torchaudio0.12.1 -f https://download.pytorch.org/whl/torch_stable.html
但是，torch和torchvision版本必须匹配
pip install torch1.10.1+cu102 torchvision==0.11.2+cu102 -f https://download.pytorch.org/whl/torch_stable.html

验证cuda与gpu是否连接成功？