Pytorch框架

环境搭建

cuda安装

查看显卡驱动对应的CUDA版本

 

 

可知为cuda9.0

下载地址 https://developer.nvidia.com/cuda-90-download-archive

安装后

cuddn安装

下载地址 https://developer.nvidia.com/rdp/cudnn-download

Pytorch环境

官网 https://pytorch.org/

下载依赖库

https://download.pytorch.org/whl/torch_stable.html

 

python虚拟环境

conda create -n pytorch_gpu python=3.6

conda activate pytorch_gpu

pip install torchvision-0.3.0-cp36-cp36m-win_amd64.whl

pip install torch-1.1.0-cp36-cp36m-win_amd64.whl