创建新环境:
1、创建conda环境
conda create -n env_name python=2.7
conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
2、加快下载速度
pip install -r requirements.txt -i http://pypi.douban.com/simple/ --trusted-host pypi.douban.com
删除虚拟环境删除环境:
使用命令conda remove -n your_env_name(虚拟环境名称) --all, 即可删除。
conda remove -n your_env_name --all
-i https://pypi.tuna.tsinghua.edu.cn/simple
import torch
print(torch.__version__)
print(torch.cuda.is_available())
判断一个数据是不是数组numpy
print(isinstance(batch, np.ndarray))
判断一个数据是不是张量 Tensor
print(isinstance(batch, torch.Tensor))
if not os.path.exists(save_file):
with open(save_file,“w”) as file:
print(“文件已创建”)
else:
print(“文件已存在”)