两种解决方法
1.法1
直接pip uninstall numpy
在pip install numpy
不用conda是因为用conda卸载numpy会把torch也卸载掉
2.法2
升级python到3.6.5
conda install python==3.6.5
我是用了上面两个方法解决的
测试是否OK
在命令行输入python
进入python之后,import torch就行
然后看看GPU是否可用:
- torch.cuda.is_available() %cuda是否可用;
- torch.cuda.device_count() %返回gpu数量;
- torch.cuda.get_device_name(0) %返回gpu名字,设备索引默认从0开始;
- torch.cuda.current_device() %返回当前设备索引;