![](https://img-blog.csdnimg.cn/20201014180756780.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Pytorch
xm_ai
这个作者很懒,什么都没留下…
展开
-
Ubuntu运行代码后台挂起命令
台挂起命令 CUDA_VISIBLE_DEVICES=0,1 nohup python -u main.py --config ./configs/train_stgan.yaml > main.log 2>&1 & CUDA_VISIBLE_DEVICES=0,1 指定显卡 main.py --config ./configs/train_stgan.yaml 指定运行.py文件 main.log 指定生成log名为main 终止后台命令 查看显卡所运行程序id nvidi原创 2021-07-17 14:22:58 · 914 阅读 · 0 评论 -
记运行CycleGAN代码遇到的问题
附上代码网址: https://github.com/aitorzip/PyTorch-CycleGAN 报错信息: requests.exceptions.ConnectionError: HTTPConnectionPool(host='localhost', port=8097): Max retries exceeded with url: /events (Caused by NewConnectionError('<urllib3.connection.HTTPConnection obj原创 2020-10-19 15:59:18 · 1263 阅读 · 2 评论 -
MNIST手写数字集实现pytorch+windows+cpu
forward下面看代码: import tensorflow as tf from tensorflow import keras from tensorflow.keras import datasets import os os.environ['TF_CPP_MIN_LOG_LEVEL'] = '2' # x:[60k,28,28] # y:[60k] (x,y),_ = dataset...原创 2019-11-04 14:45:23 · 240 阅读 · 0 评论 -
pytorch安装windows10+cpu+引入pycharm
一、创建环境+激活环境 1、创建环境 conda create -n pytorch37 python=3.7 pytorch37为环境名,python=3.7为已安装anaconda的版本号,在这里我遇到了问题,就是创建环境之后只显示done,然后什么反应都没有,后来才知道进入了cmd之后,路径是在C:\Users\吃肉肉>之下,而anaconda安装路径在D:\anaconda,在切换...原创 2020-03-13 19:33:09 · 1003 阅读 · 0 评论 -
linux下pytorch安装(cuda9.0+cudnn7.0)
前言:在服务器上安装pytorch 1、检查驱动+CUDA+CUDNN 1、在安装pytorch之前,我知道我的服务器上已经安装了驱动,查看驱动命令 nvidia-smi 如果是这样,证明已经安装了驱动 2、检查安装CUDA命令 nvcc -V 可以看到结果CUDA版本9.0 3、查看安装CUDNN命令 cat /usr/local/cuda/include/cudnn.h | grep ...原创 2020-01-30 20:23:32 · 4008 阅读 · 0 评论