pytorch的简单工程实现，数据集采用的是mnist，带有单卡和分布式多卡示例. 其中train.py，里面环境变量os.environ ["CUDA_VISIBLE_DEVICES"] = '0,1'，设置为多张卡，则执行sh multi_gpu_train.sh进行训练,注意修改.sh里面的nproc_per_node为卡的数量; 如果设置为单卡，执行python train.py即可.

PyTorch Classification Project A sample classification project using PyTorch, featuring hyperparameter tuning with grid search.

The PyTorch 1.0 release candidate introduces Torch Script, a Python subset that can be JIT-compiled into C++ or other high-speed code An official release candidate of PyTorch 1.0, the Python ...

PyTorch is an open source, machine learning framework used for both research prototyping and production deployment. According to its source code repository, PyTorch provides two high-level features: ...