服务器下载pytorch怎么看CUDA版本

您可以使用以下命令查看您的CUDA版本:

cat /usr/local/cuda/version.txt

如果您的CUDA版本为10.1，则您可以使用以下命令安装相应版本的pytorch:

pip3 install torch torchvision -f https://download.pytorch.org/whl/cu101/torch_stable.html

如果您的CUDA版本为其他版本，则需要将上面的“cu101”替换为相应的版本号，例如“cu92”或“cu100”。