![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
pytorch
毛球饲养员
这个作者很懒,什么都没留下…
展开
-
几个常用线程的
1OMP_NUM_THREADS=1export OMP_NUM_THREADS在服务器上跑PyTorch 程序的列位不要把cpu占满 修改这个环境变量降低你们的cpu使用率 (因为pytorch默认是能用多少线程就用多少的)在python代码中:# 选择显卡os.environ["CUDA_VISIBLE_DEVICES"] = "0"# 只使用一个线程os.en...原创 2017-10-08 10:56:26 · 4326 阅读 · 0 评论 -
linux github安装 Pytorch
linux 安装 pytorch原创 2017-09-29 16:10:26 · 11353 阅读 · 3 评论 -
Pytorch Conv2d
参数: Args: in_channels (int): Number of channels in the input image out_channels (int): Number of channels produced by the convolution kernel_size (int or tuple): Size of the原创 2017-11-18 19:05:10 · 7800 阅读 · 2 评论