https://github.com/331dala/cifar10_proj
Image classification CNN in PyTorch and TensorFlow.
https://github.com/331dala/cifar10_proj
cifar10 cnn jupyterlab python pytorch tensotflow
Last synced: 3 months ago
JSON representation
Image classification CNN in PyTorch and TensorFlow.
- Host: GitHub
- URL: https://github.com/331dala/cifar10_proj
- Owner: 331Dala
- Created: 2025-03-22T13:43:52.000Z (over 1 year ago)
- Default Branch: main
- Last Pushed: 2025-04-18T06:46:14.000Z (about 1 year ago)
- Last Synced: 2025-06-10T18:49:38.767Z (about 1 year ago)
- Topics: cifar10, cnn, jupyterlab, python, pytorch, tensotflow
- Language: Jupyter Notebook
- Homepage:
- Size: 2.38 MB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# cifar10_proj
Image classification CNN in PyTorch and TensorFlow.
# description
Env: Jupyter Lab
Note: Some code be comment out.
###compatibility issues
Using CUDA 11.3 and the corresponding PyTorch will encounter some Numpy errors
when running the import syntax and getting the dataset, such as train_data[0].
If you try to install the latest CUDA 12.8 on Windows 10, you will encounter problems because PyTorch doesn't support CUDA 12.8 yet.
Neither in Pip or Conda package can't find a support version.
I recommend you install CUDA 11.8, which has lots of solutions on the Internet and also supports TensorFlow.
推荐使用 TensorFlow 2.12.0，因为这个版本已官方支持 CUDA 11.8