- 下载地址:Anaconda
- 创建虚拟环境:Conda虚拟环境管理
- conda create -n torch_12 python=3.6
- 上方的torch_12为虚拟环境名,可自定义
- Cuda下载地址:Cuda10.0
- Cudnn下载地址(需要登录):Cudnn
注意:Cudnn的版本需要与Cuda对应
- Cuda安装
- 运行exe安装包
- 自定义安装
- 安装Cuda即可(驱动等其他东西一般windows已经安装了最新版)
- 安装Cudnn
- 解压zip压缩包,复制所有文件到Cuda安装目录下,如:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.0
- 验证安装:
-
Cuda: nvcc -V
-
Cudnn: nvidia-smi
-
-
安装命令查询:pytorch安装命令查询
-
进入虚拟环境:conda activate torch_12
-
安装:conda install pytorch1.1.0 torchvision0.3.0 cudatoolkit=10.0 -c pytorch
-
验证(输出true表示安装成功)
# 输入python命令进入python环境 import torch import torchvision print(torch.cuda.is_available())



