写个博客提醒一下自己,3090不能使用cuda11.0以下的版本,conda install -c pytorch cudatoolkit=11.1。
pip install torch1.8.0+cu111 torchvision0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
然后就能用了呜呜呜
写个博客提醒一下自己,3090不能使用cuda11.0以下的版本,conda install -c pytorch cudatoolkit=11.1。
pip install torch1.8.0+cu111 torchvision0.9.0+cu111 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple
然后就能用了呜呜呜