Windows 下的Pytorch +CUDA10+cudnn7的搭建
-
- pytorch 下载网址;
到官网上根据自己的电脑的配置,选择对应的参数之后,会给出对应的运行的命令;
- pytorch 下载网址;
-
- 检查是否安装成:
ipython
import torch
torch.__version__ #显示torch版本信息
torch.cuda.is_available() #是否可以使用cuda加速
ipython
import torch
torch.__version__ #显示torch版本信息
torch.cuda.is_available() #是否可以使用cuda加速