Create a pytorch (gpu) virtual environment and open it in jupyter notebook

1. Create a virtual environment

conda create -n pytorch python=3.9.7

2. Install pytorch according to the version of your graphics card

Official website link PyTorch

Select your own configuration, and then copy the code directly. If you need a past version, click the past version in the lower left corner.

My own version is 11.4, installed 11.3.

3. Install jupyter notebook

conda install ipykernel
python -m ipykernel install --name labelme

I didn't have any problems during my own installation, I will update if there are other problems.

additional:

During use, the installation of missing libraries encountered:

#matplotlib
conda install matplotlib

#cv2
pip3 install opencv-python