win10安装GPUpytorch
- 首先查看自己的python版本
激活要安装环境,输入python --version
2.然后查看CUDA版本
NVIDIA控制面板->系统信息->组件
3.然后去开源镜像文件下载相关版本的包.
4.安装下载包
在激活的环境中执行命令conda install --offline pytorch-1.7.0-py3.7_cuda102_cudnn7_0.tar.bz2
5.验证安装是否成功
import torch
torch.cuda.is_available()
返回True表示安装成功~