- 问题
使用pytorch官网的安装命令安装的pytorch可能只是支持CPU版本的
本文通过下载安装包的方式安装支持GPU的pytorch。 - 下载安装包
在pytorch开源官网https:download.pytorch.org/whl/torch_stable.html里找到对应的安装包,注意需要下载torch和torchvision两个安装包
- 安装
下载后上传到机器上,在目录下执行使用pip install安装即可
在执行命令后加上清华源,不加可能因网络超时而导致安装失败 - 验证
pip list可以看到torch和torchvision安装的为gpu版本
0条评论