PytorchGPU安装（python3.7.7+CUDA10.0+cuDNN7.6.5+pytorchgpu1.0.1）（续前）

上篇博客：https://blog.csdn.net/m0_45117053/article/details/106258915

torch和tensorflow的GPU版本安装操作之前的都一样，不同的在于是pytorch安装

torch和torchvison在https://download.pytorch.org/whl/torch_stable.html下载

注意：要找与自己的CUDA以及python版本对应的安装包
我的CUDA10.0和python3.7安装命令：

pip install C:\Users\Administrator\Downloads\torch-1.0.1-cp37-cp37m-win_amd64.whl
pip install C:\Users\Administrator\Downloads\torchvision-0.4.0-cp37-cp37m-win_amd64.whl

cp37即对应python3.7，同理cp27对应python2.7cu100即对应CUDA10.0。
在Anaconda Prompt验证是否安装成功：

这里可能遇到一个小bug，运行程序时报错

ImportError: cannot import name 'PILLOW_VERSION' from 'PIL'

解决：https://blog.csdn.net/Lee_lg/article/details/103901632