上篇博客: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'