确保环境可以运行cuda版本的pytorch
1.使用nividia-smi确保有cuda。
根据cuda版本号选择对应的pytorch。
Previous PyTorch Versions | PyTorch
pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102
1.使用nividia-smi确保有cuda。
根据cuda版本号选择对应的pytorch。
Previous PyTorch Versions | PyTorch
pip3 install torch==1.10.1+cu102 torchvision==0.11.2+cu102