pytoch 安装为Gpu版本

1，首先确保自己有显卡，然后根据显卡安装cuda ,nvidia

2,cmd 打开终端

查看自己的python版本

在下面的链接中找到相应版本的torch.torchvision 

https://download.pytorch.org/whl/torch_stable.html

一定要确保自己的torch与torchvision对应

下面是一个对应表

 

 Gpu版本是cu开头的，

 

 

 然后直接pip install 文件名.whl即可，如果出现以下这种情况，在最后面加入--user重新安装（因为权限不够）

最后安装成功！

 

测试一下是否安装成功

在cmd终端输入以下代码段

python
import torch
torch.cuda.is_available()

 显示True 则安装成功