ubuntu查看gpu及cuda版本

2 篇文章 0 订阅
1 篇文章 0 订阅

1.nvidia-smi查看gpu信息

root@jilitili:/home/jilitili# nvidia-smi
Mon Jul 19 12:58:24 2021       
+-----------------------------------------------------------------------------+
| NVIDIA-SMI 450.80.02    Driver Version: 450.80.02    CUDA Version: 11.0     |
|-------------------------------+----------------------+----------------------+
| GPU  Name        Persistence-M| Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  Quadro P2000        Off  | 00000000:21:00.0  On |                  N/A |
| 53%   46C    P0    18W /  75W |    314MiB /  5053MiB |      2%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
                                                                               
+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      1384      G   /usr/lib/xorg/Xorg                158MiB |
|    0   N/A  N/A      3191      G   compiz                             72MiB |
|    0   N/A  N/A      3414      G   fcitx-qimpanel                      6MiB |
|    0   N/A  N/A      3922      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      4169      G   ...AAAAAAAAA= --shared-files       63MiB |
|    0   N/A  N/A      5226      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      5330      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      5493      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      5625      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      5714      G   /usr/lib/firefox/firefox            1MiB |
|    0   N/A  N/A      5943      G   /usr/lib/firefox/firefox            1MiB |
+-----------------------------------------------------------------------------+
WARNING: infoROM is corrupted at gpu 0000:21:00.0
root@jilitili:/home/jilitili# 

2.查cuda版本

root@jilitili:/home/jilitili# cat /usr/local/cuda-10.0/version.txt 
CUDA Version 10.0.130
root@jilitili:/home/jilitili# cat /usr/local/cuda/version.txt 
CUDA Version 10.0.130
root@jilitili:/home/jilitili# 

3.关于GPU和显存

显卡包括GPU和显存等,GPU是计算单元,功能类似于CPU;显存是存储,类似于内存,显存越大,能够运行的模型网络和数据量越大。

关注神经网络的模型、数据、网络。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: 很高兴回答您的问题,Ubuntu Anaconda安装PyTorch CUDA=11.6 GPU版本的步骤如下:1. 安装CUDA 11.6;2. 安装Anaconda;3. 通过Anaconda安装PyTorch;4. 安装CUDA Toolkit;5. 激活GPU版本PyTorch。 ### 回答2: 要在Ubuntu上安装PyTorch CUDA 11.6 GPU版本,可以按照以下步骤进行: 1. 首先,确保您的系统中已经安装了合适版本的NVIDIA驱动程序,以支持CUDA 11.6。可以使用以下命令检查已安装的NVIDIA驱动程序版本: ``` nvidia-smi ``` 2. 接下来,安装Anaconda来管理Python环境。您可以从Anaconda官方网站下载适用于Ubuntu的Anaconda安装程序。下载完毕后,运行以下命令安装Anaconda: ``` bash Anaconda-latest-Linux-x86_64.sh ``` 3. 安装完成后,更新Anaconda: ``` conda update --all ``` 4. 创建一个新的conda环境,并激活该环境: ``` conda create --name myenv conda activate myenv ``` 5. 添加conda源,以便安装PyTorch和CUDA 11.6: ``` conda config --add channels conda-forge conda config --add channels pytorch ``` 6. 安装PyTorch和CUDA 11.6: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge ``` 7. 安装完成后,您可以使用以下命令验证是否成功安装: ``` python import torch print(torch.cuda.is_available()) ``` 如果输出为True,则表示PyTorch与CUDA 11.6 GPU版本已成功安装在您的Ubuntu系统上。 请注意,安装过程可能需要一些时间,并且根据您的系统配置以及网络连接状况,可能会有一些额外的依赖项需要安装。您可以根据安装提示进行相应的操作。 ### 回答3: 要在Ubuntu上安装PyTorch的CUDA 11.6 GPU版本,可以按照以下步骤进行操作: 1. 首先,在Ubuntu上安装Anaconda,可以从Anaconda官方网站下载适用于Linux的Anaconda安装程序。 2. 下载安装完成后,在终端中输入以下命令以创建并激活一个新的Anaconda环境: ``` conda create --name pytorch_env conda activate pytorch_env ``` 3. 接下来,安装CUDA 11.6驱动程序。可以从NVIDIA官方网站下载并根据说明进行安装。确保所使用的CUDA版本与显卡兼容。 4. 在Anaconda环境中安装PyTorch和必需的依赖项。可以使用conda命令或pip命令进行安装。以下是使用conda安装的示例命令: ``` conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch ``` 或者,使用pip命令安装: ``` pip install torch==1.9.0+cu116 torchvision==0.10.0+cu116 torchaudio==0.9.0 -f https://download.pytorch.org/whl/torch_stable.html ``` 这将安装PyTorch 1.9.0版本及其相关软件包,并与CUDA 11.6版本进行连接。 5. 安装完成后,可以编写和运行使用PyTorch进行GPU计算的程序了。可以在Python脚本中导入PyTorch库,并使用GPU进行计算。示例代码如下: ```python import torch # 检查CUDA是否可用 if torch.cuda.is_available(): device = torch.device("cuda") # 将张量分配给GPU x = torch.tensor([1.0, 2.0]).to(device) y = torch.tensor([3.0, 4.0]).to(device) z = x + y print(z) else: print("CUDA不可用") ``` 以上是在Ubuntu上使用Anaconda安装PyTorch CUDA 11.6 GPU版本的一般步骤。具体的安装细节可能会根据系统的配置和要求有所不同,请确保根据实际情况进行相应调整。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值