查看虚拟环境
conda env list
创建虚拟环境
conda create -n your_env_name python=3.x
删除某虚拟环境
conda env remove --name your_env_name
查看torch能否正常使用cuda
python
import torch
print(torch.cuda.is_available())
print(torch.backends.cudnn.is_available())#true即可
print(torch.cuda_version)
print(torch.backends.cudnn.version())#true即可
在用pytorch訓練模型時,常出現訓練結束顯存沒有釋放的問題
解決方法:
nvidia-smi 查看PID根據顯存佔用找到對應進程編號id
kill -9 [id]即可