search

WongKinYiu / yolov7

Implementation of paper - YOLOv7: Trainable bag-of-freebies sets new state-of-the-art for real-time object detectors
GNU General Public License v3.0
13.38k stars 4.23k forks source link

pytorch 1.12.1 Can I install it ？ #603

Open monkeycc opened 2 years ago

monkeycc commented 2 years ago

pytorch 1.12.1
Can I install it ？

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

cuda 11.6 cudnn 8.4

Kannan665 commented 2 years ago

Why don't you try docker? https://github.com/datamachines/cuda_tensorflow_opencv Datamachines github page has a recent notification about Pytorch 1.12.1 (20220815: Update to OpenCV 4.6.0 and PyTorch 1.12.1)....I have been using their pre-built docker images for almost 6 months, and touchwood no issues..... Before switching to docker, I was using Conda for all my Python/AI/DL environments....It make a big difference, when I switched to docker and particularly the docker images from datamachines....... https://hub.docker.com/r/datamachines/cudnn_tensorflow_opencv/tags docker pull datamachines/cudnn_tensorflow_opencv:11.6.2_2.9.1_4.6.0-20220815.....I think, this docker image has all what you are mentioning in your OP (Pytorch 1.12.1, Cuda 11.6, cuDNN 8.4)......