cnstark / pytorch-docker Goto Github PK
View Code? Open in Web Editor NEWPure Pytorch Docker Images.
License: Apache License 2.0
Pure Pytorch Docker Images.
License: Apache License 2.0
Can you publish the dockerfile for subsequent customization?
Thanks for your great work!
I am using the following docker image:
cnstark/pytorch:1.12.0-py3.9.12-cuda11.6.2-ubuntu20.04
I can not found the cuda path. /usr/local/cuda not exisits. As onnxruntime, TensorRT need the cuda well installed, could you provide more informations about this?
Hi,
Im getting this error:
Applied providers: ['CPUExecutionProvider'], with options: {'CPUExecutionProvider': {}}
find model: ./checkpoints/models/buffalo_l/w600k_r50.onnx recognition ['None', 3, 112, 112] 127.5 127.5
set det-size: (320, 320)
2024-07-06 10:56:25.700705931 [E:onnxruntime:Default, provider_bridge_ort.cc:1745 TryGetProviderInfo_CUDA] /onnxruntime_src/onnxruntime/core/session/provider_bridge_ort.cc:1426 onnxruntime::Provider& onnxruntime::ProviderLibrary::Get() [ONNXRuntimeError] : 1 : FAIL : Failed to load library libonnxruntime_providers_cuda.so with error: libcurand.so.10: cannot open shared object file: No such file or directory
Thanks.
Are you planning to add pytorch 2.0 support? Thanks for this great collection of scripts.
To my understanding some of the awkward torch==<VERSION><SUFFIX> can simply be replaced by torch==<VERSION>, while torch automatically detects CUDA and installs it torch with gpu support
hi,[昊長大人]:
pytorch的版本里面,没有找到支持cuda11.2的,导致生成sh失败,看看是否可以添加更新?
Thx
The image size of "1.13.1-py3.9.16-cuda11.7.1-devel-ubuntu20.04" on my server is 10.6 GB, while the size of that displayed on your git is 5.69 GB?
Could you add this image? thank you
rt
Could you add a new build pytorch_1.13.1_py3.9.16_cuda11.7.1_ubuntu22.04, GPU runtime and pure CPU two images.
Anyway, thank you very much.
我们一台服务器有多人使用,所以原主机的CUDA版本是11.1比较低。我可以直接拉取比较高CUDA版本的镜像使用嘛,比如CUDA11.6。使用的是A100GPU,这样会和原主机不兼容嘛。
你好,安装完之后 nvcc -V
显示了 cuda 版本,但是 nvidia-smi
命令返回错误 command not found
可以增加jupyter功能么?
Thanks for your great work!
I have docker pull some of your images, meeting the issue : sudo, vim, as well as vi command not found. Could you have a check? You can test the latest one : cnstark/pytorch:1.12.1-py3.9.12-cuda11.6.2-ubuntu20.04.
looking forward to your reply
A declarative, efficient, and flexible JavaScript library for building user interfaces.
🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.
TypeScript is a superset of JavaScript that compiles to clean JavaScript output.
An Open Source Machine Learning Framework for Everyone
The Web framework for perfectionists with deadlines.
A PHP framework for web artisans
Bring data to life with SVG, Canvas and HTML. 📊📈🎉
JavaScript (JS) is a lightweight interpreted programming language with first-class functions.
Some thing interesting about web. New door for the world.
A server is a program made to process requests and deliver data to clients.
Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.
Some thing interesting about visualization, use data art
Some thing interesting about game, make everyone happy.
We are working to build community through open source technology. NB: members must have two-factor auth.
Open source projects and samples from Microsoft.
Google ❤️ Open Source for everyone.
Alibaba Open Source for everyone
Data-Driven Documents codes.
China tencent open source team.