-
目前pytorch最新版本是1.8.0的。
-
我的显卡的CUDA版本是11,所以这里我使用:
conda create -name tc python=3.8
conda install pytorch=1.8.0 torchvision=0.9.0 torchaudio=0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
如果没有GPU:
conda install pytorch=1.8.0 torchvision=0.9.0 torchaudio=0.8.0 cpuonly -c pytorch
conda install jupyter notebook