赞
踩
我发现用上文中的安装方式,安装的pytorch版本为CPU版的,在运行程序时很卡,很影响电脑性能。
import torch
print(torch.__version__)
显示结果:2.2.1+cpu
不用考虑卸载之前版本的pytorch,在安装gpu版本时,系统会自动删除cpu版本
不建议使用镜像安装,我之前安装都不成功,直接使用pytorch官网生成的命令就可以啦
conda install pytorch torchvision torchaudio cudatoolkit=10.1 -c pytorch
这里安装pytorch要和cuda版本相匹配,若cuda版本是11.3,直接将cudatoolkit=10.1改为cudatoolkit=11.3即可。
安装成功再次查看pytorch版本不显示CPU,显示结果为:2.2.1
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。