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

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

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

显卡是没问题的，主要还是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

标签云

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

站点统计

本站现有博文242篇,共被浏览290310次

本站已经建立1778天!

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

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

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