插个note:我的电脑上在安装下面几种pytorch时用的语句如下:
python=3.5时:安装pytorch时用的语句 # CUDA 10.2
conda install pytorch==1.5.1 torchvision==0.6.1 cudatoolkit=10.2 torchaudio==0.5.0 -c pytorch -c conda-forge
python=3.7时,安装pytorch用的语句 # CUDA 11.3 ,我电脑上的rdkit用这个
conda install pytorch==1.10.0 torchvision==0.11.1 torchaudio==0.10.0 cudatoolkit=11.3 -c pytorch -c conda-forge