安装pytorch
大概配置如下图所示
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
使用anaconda会比pip快很多
安装成功
输入代码,查看torch能否使用显卡
答案是不可以
大概配置如下图所示
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
使用anaconda会比pip快很多
安装成功
输入代码,查看torch能否使用显卡
答案是不可以