conda create -n 1 python=3.6
conda activate
conda install pytorch1.8.0 torchvision0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
torch.cuda.is_available()
1.这里pytorch和cudatoolkit版本对应关系:
2.cudatoolkit版本和系统cuda对应关系:
3.系统cuda和nvidia对应关系:
4.cuda和cuDNN的关系和对应关系: