1、从pytorch官网获取安装命令

https://pytorch.org/

 

2、验证是否安装成功

进入python:

import torch 
torch.cuda.is_available()

如果返回true,则成功。

参考:https://blog.csdn.net/gdymind/article/details/82153643

更多推荐