RTX3060 CUDA11.1
安装cuda11.1 torch命令
pip install torch1.8.0+cu111 torchvision0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
查看pytorch是否装好
python
import torch
print(torch.version)
RTX3060 CUDA11.1
安装cuda11.1 torch命令
pip install torch1.8.0+cu111 torchvision0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html
查看pytorch是否装好
python
import torch
print(torch.version)