Python-Pytorch编码习惯
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