安装GPU版本pytorch

安装GPU版本pytorch

@安装GPU版本pytorch

1、torch个版本

根据自己电脑的配置下载相应的cuda和cudnn后进入链接下载自己电脑python对应版本的torch和torchvision
torch各个版本下载链接
链接中GPU版本torch和torchvision如下所示,根据需求下载：

torch：

torchvision：

2、加载

下载好后使用 pip install 文件夹位置+文件名，依次加载torch+torchvision
例如：

pip install E:torch-1.1.0-cp37-cp37m-win_amd64.whl
pip install E:torchvision-0.4.0-cp37-cp37m-win_amd64.whl

3、jupyter无内核解决：

1.conda install ipykernel
2. python -m ipykernel install --user --name torch --display-name “Python 3.7 (torch)”
（name名改成虚拟环境名称）