下载torch gpu版本
给大家分享一个安装torch的gpu安装版本步骤(正常pip安装只会安装cpu版本)
直接pip安装torch:cpu
pip 直接安装 torch
会默认安装torch的cpu版本,只能使用CPU去跑程序
安装gpu版本的torch
- 去pytorch官网找到自己电脑对应的配置.,复制对应的下载链接
- 在自己环境下安装
这里大家千万不要加下载源
否则安装不上
- 检查是否安装成功
给大家分享一个安装torch的gpu安装版本步骤(正常pip安装只会安装cpu版本)
pip 直接安装 torch
会默认安装torch的cpu版本,只能使用CPU去跑程序
安装gpu版本的torch
这里大家千万不要加下载源
否则安装不上