首先要注意不要直接使用pip install torch,会报错。
打开PyTorch官网:https://pytorch.org/get-started/locally/
根据自己的需要对操作系统、安装版本及安装工具进行选择,网页会自动生成安装命令,打开命令行输入安装命令。
安装完成后可使用如下代码进行调试
import torch
print(torch.__version__) #torch版本
print('gpu:', torch.cuda.is_available()) #gpu加速是否可用
pip工具的安装使用可参考博客:https://blog.csdn.net/sinat_40471574/article/details/91354263