nvidia-smi 查看显卡cuda 版本
nvcc --version 查看nvcc版本
Torch.cuda.is_availabel() 判断cuda正确安装
、、、、
conda uninstall cpuonly 会自动装匹配的 gpu版本
conda clean -p //删除缓存
conda clean --all
清华镜像里不一定有conda对应的 pytorch 版本,就会自动转成cpu版本安装
、、、、、、
显示显卡信息:
- import torch.cuda
- device = torch.device("cuda")
- memory_size = torch.cuda.get_device_properties(device).total_memory
- print("显存大小:", memory_size)
- frequency = torch.cuda.get_device_properties(device).clock_rate
- print("核心频率:", frequency)
CONDA 可以直接装 cuda
用该方法安装cuda和cudnn可以在不同环境中安装不同版本的cuda和cudnn,缺点是版本少且更新慢
codna search cudatoolkit
列表中选择合适的驱动
conda install cudatoolkit=11.3.1
列表中选择合适的cudnn
conda search cudnn