电脑本地安装了cuda11.2版本，显卡驱动配置如下图：
去官网查询安装pytorch-GPU版本命令：Previous PyTorch Versions | PyTorch
选择红框版本安装命令
pip install torch==1.9.1+cu111 torchvision==0.10.1+cu111 torchaudio==0.9.1 -f https://download.pytorch.org/whl/torch_stable.html
安装中
测试是否安装成功
win10系统安装pytorch-GPU环境,亲测好用
安装中
猜你喜欢
转载自blog.csdn.net/li1873997/article/details/130485013
今日推荐
周排行