Python-Pytorch编码习惯

坐吃山猪2025-07-24 13:54

Python-Pytorch编码习惯

日常开发中，关于Pytorch编码习惯的记录；更新时间：2025-07-23

1-CUDA+Pytorch安装

复制代码

# 1-Anacanda使用Python3.9
conda create -n GAN3.9 python=3.9
conda activate GAN3.9


# 2-使用cudatoolkit=11.8
conda install pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia
conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch

# 3-安装所需依赖包
pip install matplotlib 

# 4-查看GPU使用命令
nvidia-smi
watch -n 1 nvidia-smi

2-Matlab使用

兼容中文

复制代码

import matplotlib.pyplot as plt

# 兼容中文（Windows版本）
plt.rcParams['font.family'] = 'sans-serif'  # 告诉 matplotlib 使用 sans-serif 族
plt.rcParams['font.sans-serif'] = ['SimHei']  # 优先用黑体（Win 自带）
plt.rcParams['axes.unicode_minus'] = False  # 解决负号 '−' 显示成方块的问题

3-AI插件-TONGYI Lingma

安装参考网址：https://juejin.cn/post/7481858719829164083