首先我用的是VMware，ubuntu16.04.

首先建议安装anaconda,登录官网Free Download | Anaconda

下载完成后，来到安装文件目录处，打开终端，

然后在终端输入bash <anaconda文件名>

然后就一直enter和yes到底，直到安装成功

小记录：如何在虚拟机中查看显卡

输入命令 lspci | grep -i vga

接下来是安装pytorch，需要会点科学上网，否则就使用镜像网站资源。我这边就选择科学上网了。

来到官网

PyTorch

点击get started

conda install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

 然后在终端使用conda list命令，若报错，未找到该命令

就输入一下命令

echo 'export PATH="/root/anaconda3/bin:$PATH"' >> ~/.bashrc
source ~/.bashrc

 再输入conda -V

若正确显示版本，则没有问题

然后输入pytorch安装命令即可，一直enter和yes即可

安装完成后输入

python

import torch

print(torch.__version__)

若正确显示版本号，则安装成功

接下来如何在vscode中 使用conda的python解释器，而不使用vscode扩展中安装的python解释器呢

首先按下快捷键ctrl+P，然后输入> select interpreter

选择第二个即可

print(torch.__version__)
print(torch.version.cuda)#cuda版本
print(torch.backends.cudnn.version())
print(torch.cuda.is_available()) #cuda是否可用，返回为True表示可用
print(torch.cuda.device_count())#返回GPU的数量
print(torch.cuda.get_device_name(0))#返回gpu名字，设备索引默认从0开始