问题: Disable or able cudnn,查询版本. Disable cudnn for batch_norm: (See: @Microsoft / human-pose-estimation.pytorch#installation) # PYTORCH=/path/to/pytorch # for pytorch v0.4.0 sed -i "1194s/torch\.backends\.cudnn\.enabled/False/g" ${PYTORCH}/torch/
CUDA以及CUDNN下载安装 在https://developer.nvidia.com/cuda-toolkit-archive可以找到各个版本的cuda,个人建议下载cuda11.0(也就是2年前的),太新或太旧都可能出现适配问题 在https://developer.nvidia.com/rdp/cudnn-archive下载符合cuda版本的cudnn压缩包,解压之后把里面的bin.include.lib文件夹等复制到cuda安装路径下(CUDA默认安装路径为C:\Program Fi
Summary on deep learning framework --- PyTorch Updated on 2018-07-22 21:25:42 import osos.environ["CUDA_VISIBLE_DEVICES"]="4" 1. install the pytorch version 0.1.11 ## Version 0.1.11 ## python2.7 and cuda 8.0 sudo pip install http://