- 下载安装cuda/cudnn/anaconda,anaconda用自己需要的python的版本
- 百度pytorch进入官网,选择windows,cuda,python版本等,得到下载代码
- 复制下载代码中的网址,在浏览器中打开
- 找自己需要的torch和torchvision,用迅雷下载很快
- 在下载目录cmd后,输入pip install 那两个文件的名字
- 提示缺少依赖项,pip install 补上即可
- 用import torch print(torch.cuda.is_available()) 验证pytorch-gpu是否安好
windows下用pip国内源的方式安装pytorch
最新推荐文章于 2024-05-25 18:09:06 发布