Environment setup (Anaconda, Pytorch)

2022-09-14 version

Environment setup

OS: Ubuntu 20.04

CUDA: 11.6.2

cuDNN:  8.4.0.27

Anaconda: 64-Bit (x86) Installer (659 MB)

Pytorch: 1.12.1 with CUDA11.6

 

1. Anaconda

https://www.anaconda.com/products/distribution
64-Bit (x86) Installer (659 MB)

installation finished.
WARNING:
    You currently have a PYTHONPATH environment variable set. This may cause
    unexpected behavior when running the Python interpreter in Anaconda3.
    For best results, please verify that your PYTHONPATH only points to
    directories of packages that are compatible with the Python interpreter
    in Anaconda3: /home/sip2/anaconda3
Do you wish the installer to initialize Anaconda3
by running conda init? [yes|no]
[no] >>> no

sip2@sip2-2021:~$ source ~/anaconda3/etc/profile.d/conda.sh
sip2@sip2-2021:~$ conda create --name py38-test python=3.8

#
# To activate this environment, use
#
#     $ conda activate py38-test
#
# To deactivate an active environment, use
#
#     $ conda deactivate

2. Pytorch1.12.1

sip2@sip2-2021:~$ conda create --name py38-torch1-12-1 python=3.8
sip2@sip2-2021:~$ conda activate py38-torch1-12-1

(py38-torch1-12-1) sip2@sip2-2021:~$ conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge