PyTorch
安装
官网：
1 | https://pytorch.org/get-started/locally/?_gl=1*10s9d6u*_up*MQ..*_ga*Njk0MTY2ODU3LjE3NzkxODk1ODI.*_ga_469Y0W5V62*czE3NzkxODk1ODEkbzEkZzEkdDE3NzkxODk3MzgkajI1JGwwJGgw |
自行选择想要下载的版本如果复制命令安装：
装完后，在Python里验证：
1 | import torch |
如果第二行是 True，说明PyTorch能用NVIDIA GPU。
一般可以这样设置：
1 | device = "cuda" if torch.cuda.is_available() else "cpu" |
