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控制面板-->帮助-->系统信息-->组件
开始安装:
首先将安装命令复制到conda中:
conda install pytorch torchvision torchaudio cudatoolkit=11.0 -c pytorch