torch.cuda.is_available()返回false的解决办法| 东毅居士

torch.cuda.is_available()返回false的解决办法

作者：XD / 发表： 2020年4月27日 01:51 / 更新： 2020年5月22日 02:14 / 编程笔记 / 阅读量：9670

显卡是没问题的，主要还是cudatoolkit版本的问题。

nvidia-smi 查看cuda版本

然后复制这一行就搞定了

conda install pytorch torchvision cudatoolkit=xxx(选择对应的版本) -c pytorch

本文作者：XD 转载请标明出处：http://www.eadst.com/blog/6

本站采用知识共享署名-非商业性使用-相同方式共享 4.0 国际许可协议进行许可。

上一篇
Code for SPIE paper - CEIR

下一篇
Add uploading files feature for Django admin

Category

标签云

FlashAttention LoRA OCR Hungarian Windows Data Django CEIR Augmentation SQL Ubuntu Vmess Quantize HaggingFace Plate Excel hf Michelin Jetson v2ray YOLO CV 签证 Crawler Algorithm torchinfo Tensor 搞笑 BeautifulSoup AI Tiktoken Pillow XML MD5 TensorFlow UNIX Numpy git CUDA Bipartite NLTK JSON uwsgi Diagram Statistics Ptyhon TensorRT ChatGPT 报税 EXCEL HuggingFace Disk Pickle GPT4 Base64 Clash 阿里云 Pandas Web Dataset Proxy Hotel TSV Random Hilton Sklearn git-lfs Zip Password Distillation CAM ONNX Python SVR BTC OpenCV 关于博主 Knowledge Magnet GPTQ CSV Bin InvalidArgumentError Docker Review FP32 Bitcoin Git VSCode FP16 Card Anaconda CTC Math NameSilo VPN Conda COCO Pytorch Markdown Template GIT Transformers PDF GoogLeNet Image2Text transformers C++ LLM Google Domain Streamlit Use 域名 VGG-16 Vim LeetCode NLP Tracking WebCrawler Heatmap Website Food Interview Translation uWSGI ModelScope SPIE Permission Nginx diffusers SQLite QWEN Github Safetensors Baidu Qwen2.5 tar Linux llama.cpp PIP XGBoost Paddle OpenAI RGB Cloudreve Plotly Bert Gemma logger Paper ResNet-50 Datetime FP64 Qwen 腾讯云 Attention Quantization Land Color PyCharm 算法题 LLAMA LaTeX Video BF16 FastAPI DeepStream Input GGML scipy Shortcut

站点统计

本站现有博文266篇,共被浏览440542次

本站已经建立2019天!

原 torch.cuda.is_available()返回false的解决办法

作者：XD / 发表： 2020年4月27日 01:51 / 更新： 2020年5月22日 02:14 / 编程笔记 / 阅读量：9670

torch.cuda.is_available()返回false的解决办法