pytorch

查看某个环境是否可用gpu

先进入目标环境
In [1]: import torch
In [2]: print("GPU 数量:", torch.cuda.device_count())
GPU 数量: 4
In [3]: print("PyTorch 版本:", torch.__version__)
PyTorch 版本: 2.0.0+cu118
In [4]: print("CUDA 版本:", torch.version.cuda)
CUDA 版本: 11.8
In [5]: print("是否支持 GPU:", torch.cuda.is_available())
是否支持 GPU: True
In [6]: print("当前 GPU:", torch.cuda.get_device_name(0) if torch.cuda.is_available() else "无")
当前 GPU: NVIDIA GeForce RTX 4090
Linux服务器pytorch环境配置详解_linux创建torch的虚拟环境-CSDN博客
清怡师姐：
手把手教你在linux上安装pytorch与cuda_linux系统下安装pytorch和cuda-CSDN博客
nvidia-smi,这个里面的cuda是服务器的驱动cuda,
在4090里面按照上面教程配置会报错，显示
(cyj) dddc@gpu-4090:~/chenyuanjie/pytorch_/cuda-11.8$ conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c nvidia
Solving environment: failed
CondaValueError: Malformed version string '~': invalid character(s).
channels:
  - defaults
  - conda-forge
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
  - https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/msys2/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/fastai/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/bioconda/
  - http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/rdkit/
show_channel_urls: true
ssl_verify: false
solver: classic

85.9上pytorch

pytorch1.12.1
PyTorch 版本: 1.12.1+cu113
Python 3.9.20
pytorch_gpu_9.1
PyTorch 版本: 1.3.1
Python 3.7.10
CUDA 版本: 9.2.148
85.9上我的环境
cyj,这个是pytorch没按成功gpu版本的,里面有rdkit
cyj2，成功安装pytorch_gpu版本的，里面有rdkit。20250425的rdkit出错了，重新安装配置这个环境
 conda create --name cyj2 --clone pytorch1.12.1
 conda create --name cyj_new --clone pytorch_gpu_9.1
 
  •  (cyj_new) [chpeng@localhost check_yaodian_532tcm]$ conda list |grep rdkit
    • 

  • rdkit                     2021.03.2        py37haf5a968_0    conda-forge
    • 

  • (cyj_new) [chpeng@localhost check_yaodian_532tcm]$ ipython
    • 

  • Python 3.7.10 (default, Feb 26 2021, 18:47:35)
    • 

  • Type 'copyright', 'credits' or 'license' for more information
    • 

  • IPython 7.34.0 -- An enhanced Interactive Python. Type '?' for help.
    • 


    • 

  • In [1]: from rdkit import Chem
    • 


    • 

  • In [2]: from rdkit.Chem import AllChem
    • 


    • 

  • In [3]: mol = Chem.MolFromSmiles("CCO")
    • 


    • 

  • In [4]: print(mol)
    • 

  • <rdkit.Chem.rdchem.Mol object at 0x2ad4aaa401c0>