远程控制实验室服务器运行代码时报错CUDAerror,尝试了很多网上的教程才成功,码一下最后成功的解决方法。
服务器显卡是双卡的3070,CUDA版本是11.7,虚拟环境的python版本是3.6,在激活虚拟环境的前提下输入以下代码,运行之后可以成功解决问题:
pip install torch==1.7.0+cu110 torchvision==0.8.1+cu110 torchaudio===0.7.0 -f https://download.pytorch.org/whl/torch_stable.html