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

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

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

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

标签云

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

站点统计

本站现有博文258篇,共被浏览416901次

本站已经建立1985天!

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

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

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