第一,先在该环境下创建test.py,输入下面代码
结果:
或者:
我们发现是CPU版。
第二,用管理员身份打开conda命令行,卸载cpu版的pytorch
打开命令行
PyTorch,建议原版下载。
最后在原来的test.py测试
我们成功安装了GPU版的pytorch
总结:
卸载cpu的pytorch,安装GPU的pytorch
第一,先在该环境下创建test.py,输入下面代码
结果:
或者:
我们发现是CPU版。
第二,用管理员身份打开conda命令行,卸载cpu版的pytorch
打开命令行
PyTorch,建议原版下载。
最后在原来的test.py测试
我们成功安装了GPU版的pytorch
总结:
卸载cpu的pytorch,安装GPU的pytorch