Win10-pytorch-GPU 安装

已经通过 conda install pytorch 安装过cpu版本

安装GPU版本

安装CUDA

https://developer.nvidia.com/cuda-toolkit 下载最新版本的CUDA

需要先安装visual studio（安装CUDA的过程中会提示）

https://visualstudio.microsoft.com/zh-hans/ 下载visual studio并安装

然后继续安装CUDA即可，安装结束后命令行输入 nvcc -V 成功显示CUDA信息

安装gpu版本pytorch 

安装前

    import torch

    torch.cuda.is_available()

    output: False

https://pytorch.org/get-started/locally/

conda找不到torchvision ，尝试用pip安装

pip3 install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

命令行的这条命令最后显示成功安装了torch和torchvision，torch.cuda.is_available()为True

但是在conda的虚拟环境中并没有被成功安装。

解决方法是保证这个虚拟环境下应该安装了pip，然后运行

pip install torch===1.4.0 torchvision===0.5.0 -f https://download.pytorch.org/whl/torch_stable.html

成功后

    import torch

    torch.cuda.is_available()

    output: True

参考：

https://blog.csdn.net/cmat2/article/details/80407059