博主您好,我在autodl上租用服务器运行pytracking框架,用的是RTX 3090显卡,装好环境后运行发现报错Torch not compiled with CUDA enabled,我的pytorch和对应的cuda应该是用conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch下载的,请问您用的版本是什么,可以借鉴一下吗
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch
这个主要可能是pytorch和cuda版本不匹配造成的,我是通过https://download.pytorch.org/whl/torch_stable.html这里面下载安装的。但是autodl应该是可以直接选预装好的pytorch镜像的,然后第一步进去就先试试torch可不可以用
不知道你这个问题是否已经解决, 如果还没有解决的话:在anaconda prompt输入 conda list
可以发现pytorch是CPU还是GPU版,应该下载GPU版
pip uninstall torch后
去官网
Previous PyTorch Versions | PyTorch找命令下载,一步到位