window系统:conda 安装pytorch-gpu及速度缓慢解决方式
首先建议先新建一个虚拟环境:
conda create -n pytorch-gpu python=3.8
切换到pytorch环境
conda activate pytorch-gpu
安装pytorch
进入官网https://pytorch.org/get-started/locally/
提示:这里请根据自己的显卡情况选择对应的CUDA生成!
cuda版本查看:
windows:
桌面右键-->NVDIA控制面板-->帮助-->系统信息-->组件