新的服务器CUDA版本是10.0,与torch-0.4.0,torchvision -0.2.1不匹配。(之前CUDA版本9.0 装的是这个)
出现了
UserWarning:
Found GPU0 GeForce RTX 2080 Ti which requires CUDA_VERSION >= 9000 for
optimal performance and fast startup time, but your PyTorch was compiled
with CUDA_VERSION 8000. Please install the correct PyTorch binary
using instructions from http://pytorch.org
……
OSError: [Errno 12] Cannot allocate memory。
检查内存后内存充足,且代码需要的内存很少,考虑是版本问题。
新建了python环境:conda create --name py36_torch python=3.6
装了最新版的torch(1.3.1)和torchvison(0.4.2)
pip install torch torchvision -i https://pypi.douban.com/simple
运行测试代码成功!(先安装代码需要的库)